[InstructBLIP] Fix bos token of LLaMa checkpoints (#24492)

* Add fix * Fix doctest
2023-07-11 21:43:01 +02:00
parent aac4c79968
commit bb13a92859
2 changed files with 8 additions and 3 deletions
--- a/tests/models/instructblip/test_modeling_instructblip.py
+++ b/tests/models/instructblip/test_modeling_instructblip.py
@@ -544,7 +544,6 @@ class InstructBlipModelIntegrationTest(unittest.TestCase):

        # verify generation
        outputs = model.generate(**inputs, max_new_tokens=30)
-        outputs[outputs == 0] = 2
        generated_text = processor.batch_decode(outputs, skip_special_tokens=True)[0].strip()

        # fmt: off