Fix SDPA attention precision issue in Qwen2.5-VL (#37363)
* solve conflicts and remove redundant attention_mask in qwenvit * update decoded text check * remove trailing whitespace
This commit is contained in:
@@ -419,7 +419,7 @@ class Glm4vIntegrationTest(unittest.TestCase):
|
||||
output = model.generate(**inputs, max_new_tokens=30)
|
||||
|
||||
EXPECTED_DECODED_TEXT = [
|
||||
"\nWhat kind of dog is this?\n<think>Got it, let's look at the image. The animal in the picture has a stocky build, thick fur, and a face that's",
|
||||
"\nWhat kind of dog is this?\n<think>Got it, let's look at the image. The animal in the picture is not a dog; it's a cat. Specifically, it looks",
|
||||
"\nWhat kind of dog is this?\n<think>Got it, let's look at the image. Wait, the animals here are cats, not dogs. The question is about a dog, but"
|
||||
] # fmt: skip
|
||||
self.assertEqual(
|
||||
|
||||
Reference in New Issue
Block a user