Skip to content

Commit 0c32bab

Browse files
committed
Revert "enable xpu"
This reverts commit 2ebe9ca.
1 parent 2ebe9ca commit 0c32bab

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/models/unets/test_models_unet_2d_condition.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@
4646
require_peft_backend,
4747
require_torch_accelerator,
4848
require_torch_accelerator_with_fp16,
49+
require_torch_gpu,
4950
skip_mps,
5051
slow,
5152
torch_all_close,
@@ -979,7 +980,7 @@ def test_ip_adapter_plus(self):
979980
assert sample2.allclose(sample5, atol=1e-4, rtol=1e-4)
980981
assert sample2.allclose(sample6, atol=1e-4, rtol=1e-4)
981982

982-
@require_torch_accelerator
983+
@require_torch_gpu
983984
@parameterized.expand(
984985
[
985986
("hf-internal-testing/unet2d-sharded-dummy", None),
@@ -995,7 +996,7 @@ def test_load_sharded_checkpoint_from_hub(self, repo_id, variant):
995996
assert loaded_model
996997
assert new_output.sample.shape == (4, 4, 16, 16)
997998

998-
@require_torch_accelerator
999+
@require_torch_gpu
9991000
@parameterized.expand(
10001001
[
10011002
("hf-internal-testing/unet2d-sharded-dummy-subfolder", None),

0 commit comments

Comments
 (0)