fix test

2025-12-06 20:44:33 +08:00 · 2023-09-12 15:23:07 +00:00
1 changed files with 3 additions and 1 deletions
--- a/tests/pipelines/stable_diffusion/test_stable_diffusion.py
+++ b/tests/pipelines/stable_diffusion/test_stable_diffusion.py
@@ -923,7 +923,9 @@ class StableDiffusionPipelineSlowTests(unittest.TestCase):
        torch.cuda.reset_max_memory_allocated()
        torch.cuda.reset_peak_memory_stats()

-        pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16)
+        pipe = StableDiffusionPipeline.from_pretrained(
+            "CompVis/stable-diffusion-v1-4", safety_checker=None, torch_dtype=torch.float16
+        )
        pipe = pipe.to(torch_device)
        pipe.set_progress_bar_config(disable=None)
        pipe.enable_attention_slicing(1)