fix: loss computation after embeddings resize - mllama #36840
move loss to generation class
985fe758
code cleanup
98b8b290
Ssukriti
changed the title fix loss computation after embeddings resize fix: loss computation after embeddings resize - mllama 1 year ago
test for resize and loss computation
dd02bfdf
Ssukriti
marked this pull request as ready for review 1 year ago
fix tests
aef4f515
fix:test for resize and loss
c7b9adb5
fix resize embedding mllama test
75714c6e
review changes
1d864527
Ssukriti
deleted the test_fix_loss_computation branch 1 year ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub