"comfy/git@developer.sourcefind.cn:chenpangpang/ComfyUI.git" did not exist on "f2a7cc912186c89fda9580f36da28c7fc382ea26"
Unverified Commit 5e620a92 authored by Yoach Lacombe's avatar Yoach Lacombe Committed by GitHub
Browse files

Fix `SeamlessM4Tv2ModelIntegrationTest` (#27911)

change dtype of some integration tests
parent e96c1de1
......@@ -1014,8 +1014,9 @@ class SeamlessM4Tv2ModelIntegrationTest(unittest.TestCase):
)
def factory_test_task(self, class1, class2, inputs, class1_kwargs, class2_kwargs):
model1 = class1.from_pretrained(self.repo_id).to(torch_device)
model2 = class2.from_pretrained(self.repo_id).to(torch_device)
# half-precision loading to limit GPU usage
model1 = class1.from_pretrained(self.repo_id, torch_dtype=torch.float16).to(torch_device)
model2 = class2.from_pretrained(self.repo_id, torch_dtype=torch.float16).to(torch_device)
set_seed(0)
output_1 = model1.generate(**inputs, **class1_kwargs)
......
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