-
Notifications
You must be signed in to change notification settings - Fork 52
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Fix HFTrainer overloads #869
Conversation
👋 Hi! Thank you for contributing to llm-compressor. Please add the ready label when the PR is ready for review. |
9d51505
to
9a7711e
Compare
Should we add *args, **kwargs and pass them to the super functions, to protect against future changes like this? |
@rahul-tuli Since both functions require some information from the inputs, I think I'd prefer to expose all the arguments. There are tradeoffs either way w.r.t. erroring when arguments change |
b6b54fb
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
b6b54fb
to
84be172
Compare
Purpose
tests/llmcompressor/transformers/finetune/test_finetune_no_recipe_custom_dataset.py
Background
Changes
Testing
tests/llmcompressor/transformers/finetune/test_finetune_no_recipe_custom_dataset.py
now passes