fix: loss computation after embeddings resize - mllama (#36840)
* move loss to generation class
Signed-off-by: Sukriti-Sharma4 <sukriti.sharma4@ibm.com>
* code cleanup
Signed-off-by: Sukriti-Sharma4 <sukriti.sharma4@ibm.com>
* test for resize and loss computation
Signed-off-by: Sukriti-Sharma4 <sukriti.sharma4@ibm.com>
* fix tests
Signed-off-by: Sukriti-Sharma4 <sukriti.sharma4@ibm.com>
* fix:test for resize and loss
Signed-off-by: Sukriti-Sharma4 <sukriti.sharma4@ibm.com>
* fix resize embedding mllama test
Signed-off-by: Sukriti-Sharma4 <sukriti.sharma4@ibm.com>
* review changes
Signed-off-by: Sukriti-Sharma4 <sukriti.sharma4@ibm.com>
---------
Signed-off-by: Sukriti-Sharma4 <sukriti.sharma4@ibm.com>