riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 18:22:34 +06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix vit hybrid test (#25543)

Browse Source 
fix test
This commit is contained in:
Marc Sun 2023-08-16 11:02:57 -04:00 committed by GitHub
parent 3f9cb33504
commit 0ed23e4db2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/vit_hybrid/test_modeling_vit_hybrid.py
View File

@ -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