transformers
fix: loss computation after embeddings resize - mllama
#36840
Merged

fix: loss computation after embeddings resize - mllama #36840

Ssukriti
Ssukriti move loss to generation class
985fe758
Ssukriti code cleanup
98b8b290
Ssukriti
Ssukriti commented on 2025-03-19
Ssukriti Ssukriti changed the title fix loss computation after embeddings resize fix: loss computation after embeddings resize - mllama 1 year ago
Ssukriti test for resize and loss computation
dd02bfdf
Ssukriti Ssukriti marked this pull request as ready for review 1 year ago
Ssukriti fix tests
aef4f515
Ssukriti fix:test for resize and loss
c7b9adb5
Ssukriti
Ssukriti commented on 2025-03-20
Ssukriti fix resize embedding mllama test
75714c6e
Ssukriti
Ssukriti commented on 2025-03-20
zucchini-nlp
zucchini-nlp commented on 2025-03-20
Ssukriti review changes
1d864527
Ssukriti
zucchini-nlp
zucchini-nlp approved these changes on 2025-03-21
Ssukriti
zucchini-nlp
zucchini-nlp zucchini-nlp merged 90e2df5d into main 1 year ago
Ssukriti Ssukriti deleted the test_fix_loss_computation branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone