Open MorenoLaQuatra opened 2 months ago
transformers
@muellerzr @SunMarc
examples
The issue is reported here: https://discuss.huggingface.co/t/valueerror-the-batch-received-was-empty-your-model-wont-be-able-to-train-on-it-double-check-that-your-training-dataset-contains-keys-expected-by-the-model-args-kwargs-label-ids-label/20200
Correctly working fine-tuning with LoRA. Anyway, I thing the bug can be solved adding:
self._signature_columns += list(set(["labels", "input_ids"]))
To the function _set_signature_columns_if_needed - here.
_set_signature_columns_if_needed
May not be the best way to go tho.
Gentle ping @muellerzr @SunMarc
System Info
transformers
version: 4.38.2Who can help?
@muellerzr @SunMarc
Information
Tasks
examples
folder (such as GLUE/SQuAD, ...)Reproduction
The issue is reported here: https://discuss.huggingface.co/t/valueerror-the-batch-received-was-empty-your-model-wont-be-able-to-train-on-it-double-check-that-your-training-dataset-contains-keys-expected-by-the-model-args-kwargs-label-ids-label/20200
Expected behavior
Correctly working fine-tuning with LoRA. Anyway, I thing the bug can be solved adding:
To the function
_set_signature_columns_if_needed
- here.May not be the best way to go tho.