Sukriti Sharma 90e2df5d55 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>		2025-03-21 14:47:59 +01:00
..
__init__.py	Add MLLama (#33703)	2024-09-25 19:56:25 +02:00
test_image_processing_mllama.py	Fix bugs in mllama image processing (#36156)	2025-03-11 10:22:48 +01:00
test_modeling_mllama.py	fix: loss computation after embeddings resize - mllama (#36840)	2025-03-21 14:47:59 +01:00
test_processor_mllama.py	Mllama: raise better error (#35934)	2025-03-21 09:35:37 +01:00