remove

Author: faaany

tests/single_file/test_model_vae_single_file.py

@@ -25,7 +25,7 @@
     enable_full_determinism,
     load_hf_numpy,
     numpy_cosine_similarity_distance,
-    require_torch_accelerator,
+    require_torch_gpu,
     slow,
     torch_device,
 )
@@ -35,7 +35,7 @@
 
 
 @slow
-@require_torch_accelerator
+@require_torch_gpu
 class AutoencoderKLSingleFileTests(unittest.TestCase):
     model_class = AutoencoderKL
     ckpt_path = (