Fix Mistral3 tests (#36797)

* fix processor tests

* fix modeling tests

* fix test processor chat template

* revert modeling test changes
This commit is contained in:
Yoni Gozlan
2025-03-18 13:08:12 -04:00
committed by GitHub
parent db1d4c5a0b
commit 30580f035b
2 changed files with 67 additions and 5 deletions

View File

@@ -54,7 +54,7 @@ Here is how you can use the `image-text-to-text` pipeline to perform inference w
... },
... ]
>>> pipe = pipeline("image-text-to-text", model="../mistral3_weights", torch_dtype=torch.bfloat16)
>>> pipe = pipeline("image-text-to-text", model="mistralai/Mistral-Small-3.1-24B-Instruct-2503", torch_dtype=torch.bfloat16)
>>> outputs = pipe(text=messages, max_new_tokens=50, return_full_text=False)
>>> outputs[0]["generated_text"]
'The image depicts a vibrant and lush garden scene featuring a variety of wildflowers and plants. The central focus is on a large, pinkish-purple flower, likely a Greater Celandine (Chelidonium majus), with a'