Unverified Commit 1ea3ad1a authored by Fanli Lin's avatar Fanli Lin Committed by GitHub
Browse files

[tests] use `torch_device` instead of `auto` for model testing (#29531)



* use torch_device

* skip for XPU

* Update tests/generation/test_utils.py
Co-authored-by: default avataramyeroberts <22614925+amyeroberts@users.noreply.github.com>

---------
Co-authored-by: default avataramyeroberts <22614925+amyeroberts@users.noreply.github.com>
parent 14536c33
...@@ -1073,6 +1073,9 @@ class GenerationTesterMixin: ...@@ -1073,6 +1073,9 @@ class GenerationTesterMixin:
@require_torch_multi_accelerator @require_torch_multi_accelerator
def test_model_parallel_beam_search(self): def test_model_parallel_beam_search(self):
for model_class in self.all_generative_model_classes: for model_class in self.all_generative_model_classes:
if "xpu" in torch_device:
return unittest.skip("device_map='auto' does not work with XPU devices")
if model_class._no_split_modules is None: if model_class._no_split_modules is None:
continue continue
......
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