Add kwargs to T5ForClassification forward

boun-tabi-LMG / turkish-lm-tuner

Turkish LM Tuner

https://boun-tabi-lmg.github.io/turkish-lm-tuner/

MIT License

76 stars 6 forks source link

Add kwargs to T5ForClassification forward #75

Open gokceuludogan opened 4 months ago

gokceuludogan commented 4 months ago

While using T5ForClassification with Hugging Face Trainer, additional parameters such as input_embeds may be provided, potentially causing model training failure. This PR fixes this issue by introducing kwargs to the forward method.