You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
[Bugfix][Examples] drop data_collator causing failures in trl init (#1606)
SUMMARY:
[A recent update to
trl](https://github.com/huggingface/trl/pull/3076/files) no longer
allows data_collator to be set when padding_free=True, causing
`examples/trl_mixin/ex_trl_distillation.py` to fail. This drops the
default collator, which is not needed, to allow for performant packing.
Resolves INFERENG-959
TEST PLAN:
Resolved error in `examples/trl_mixin/ex_trl_distillation.py` in local
check after this update.
---------
Signed-off-by: Brian Dellabetta <bdellabe@redhat.com>
0 commit comments