address feedback

sayakpaul · sayakpaul · commit 58b79f2c12cf · 2024-12-02T14:24:22.000+05:30
diff --git a/tests/lora/test_lora_layers_sd3.py b/tests/lora/test_lora_layers_sd3.py
@@ -36,7 +36,6 @@
     require_big_gpu_with_torch_cuda,
     require_peft_backend,
     require_torch_gpu,
-    slow,
     torch_device,
 )
 
@@ -134,7 +133,6 @@ def test_modify_padding_mode(self):
         pass
 
 
-@slow
 @nightly
 @require_torch_gpu
 @require_peft_backend
diff --git a/tests/pipelines/controlnet_flux/test_controlnet_flux.py b/tests/pipelines/controlnet_flux/test_controlnet_flux.py
@@ -35,7 +35,6 @@
     nightly,
     numpy_cosine_similarity_distance,
     require_big_gpu_with_torch_cuda,
-    slow,
     torch_device,
 )
 from diffusers.utils.torch_utils import randn_tensor
@@ -205,7 +204,6 @@ def test_flux_image_output_shape(self):
             assert (output_height, output_width) == (expected_height, expected_width)
 
 
-@slow
 @nightly
 @require_big_gpu_with_torch_cuda
 @pytest.mark.big_gpu_with_torch_cuda
diff --git a/tests/pipelines/flux/test_pipeline_flux.py b/tests/pipelines/flux/test_pipeline_flux.py
@@ -12,7 +12,6 @@
     nightly,
     numpy_cosine_similarity_distance,
     require_big_gpu_with_torch_cuda,
-    slow,
     torch_device,
 )
 
@@ -207,7 +206,6 @@ def test_flux_image_output_shape(self):
             assert (output_height, output_width) == (expected_height, expected_width)
 
 
-@slow
 @nightly
 @require_big_gpu_with_torch_cuda
 @pytest.mark.big_gpu_with_torch_cuda
diff --git a/tests/pipelines/mochi/test_mochi.py b/tests/pipelines/mochi/test_mochi.py
@@ -28,7 +28,6 @@
     numpy_cosine_similarity_distance,
     require_big_gpu_with_torch_cuda,
     require_torch_gpu,
-    slow,
     torch_device,
 )
 
@@ -263,7 +262,6 @@ def test_vae_tiling(self, expected_diff_max: float = 0.2):
         )
 
 
-@slow
 @nightly
 @require_torch_gpu
 @require_big_gpu_with_torch_cuda

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,6 @@`
`28`	`28`	`numpy_cosine_similarity_distance,`
`29`	`29`	`require_big_gpu_with_torch_cuda,`
`30`	`30`	`require_torch_gpu,`
`31`		`- slow,`
`32`	`31`	`torch_device,`
`33`	`32`	`)`
`34`	`33`
`@@ -263,7 +262,6 @@ def test_vae_tiling(self, expected_diff_max: float = 0.2):`
`263`	`262`	`)`
`264`	`263`
`265`	`264`
`266`		`-@slow`
`267`	`265`	`@nightly`
`268`	`266`	`@require_torch_gpu`
`269`	`267`	`@require_big_gpu_with_torch_cuda`