Skip to content

GitLab

"tests/models/starcoder2/test_modeling_starcoder2.py" did not exist on "83ab0115d1e93009eb52b66096e924bb44f928a1"
Unverified Commit 0ed23e4d authored by Marc Sun's avatar Marc Sun Committed by GitHub
Browse files

fix vit hybrid test (#25543) 

fix test
parent 3f9cb335
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/models/vit_hybrid/test_modeling_vit_hybrid.py
View file @ 0ed23e4d
...@@ -281,7 +281,7 @@ class ViTModelIntegrationTest(unittest.TestCase): ...@@ -281,7 +281,7 @@ class ViTModelIntegrationTest(unittest.TestCase):
image = prepare_img() image = prepare_img()
inputs = image_processor(images=image, return_tensors="pt") inputs = image_processor(images=image, return_tensors="pt").to(torch_device)
outputs = model(**inputs) outputs = model(**inputs)
logits = outputs.logits logits = outputs.logits
# model predicts one of the 1000 ImageNet classes # model predicts one of the 1000 ImageNet classes
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment