Skip to content

GitLab

"docs/source/vscode:/vscode.git/clone" did not exist on "76cadb7943c8492ec481f4f3925e9e8793a32c9d"
Unverified Commit 5a551df9 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix `TvpModelIntegrationTests` (#27792) 



* fix

* fix

* fix

---------
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent c0b9db09
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
tests/models/tvp/test_modeling_tvp.py
View file @ 5a551df9
......@@ -230,10 +230,11 @@ class TvpModelIntegrationTests(unittest.TestCase):
image_processor = self.default_image_processor
image = prepare_img()
encoding = image_processor(images=image, return_tensors="pt").to(torch_device)
encoding = image_processor(images=image, return_tensors="pt")
input_ids = torch.tensor([[1, 2]])
attention_mask = torch.tensor([[1, 1]])
encoding.update({"input_ids": input_ids, "attention_mask": attention_mask})
encoding.to(torch_device)
with torch.no_grad():
outputs = model(**encoding)
......@@ -250,10 +251,11 @@ class TvpModelIntegrationTests(unittest.TestCase):
image_processor = self.default_image_processor
image = prepare_img()
encoding = image_processor(images=image, return_tensors="pt").to(torch_device)
encoding = image_processor(images=image, return_tensors="pt")
input_ids = torch.tensor([[1, 2]])
attention_mask = torch.tensor([[1, 1]])
encoding.update({"input_ids": input_ids, "attention_mask": attention_mask})
encoding.to(torch_device)
with torch.no_grad():
outputs = model(**encoding)
......
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