From 5a7100b780d88978a3689dd5611a6d715949383d Mon Sep 17 00:00:00 2001 From: statelesshz Date: Thu, 4 Apr 2024 15:51:09 +0800 Subject: [PATCH] make style --- tests/test_big_modeling.py | 1 - tests/test_utils.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/test_big_modeling.py b/tests/test_big_modeling.py index a13ab197fce..9533207149d 100644 --- a/tests/test_big_modeling.py +++ b/tests/test_big_modeling.py @@ -35,7 +35,6 @@ from accelerate.test_utils import ( require_bnb, require_cuda, - require_mps, require_multi_device, require_multi_gpu, require_non_cpu, diff --git a/tests/test_utils.py b/tests/test_utils.py index 0e197471032..b992b349964 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -28,8 +28,8 @@ from accelerate.state import PartialState from accelerate.test_utils.testing import ( require_cuda, - require_non_cpu, require_huggingface_suite, + require_non_cpu, require_non_torch_xla, require_torch_min_version, require_tpu,