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