fix

Author: sayakpaul

tests/lora/test_lora_layers_flux.py

@@ -30,7 +30,6 @@
     is_peft_available,
     nightly,
     numpy_cosine_similarity_distance,
-    print_tensor_test,
     require_big_gpu_with_torch_cuda,
     require_peft_backend,
     require_torch_gpu,
@@ -195,7 +194,7 @@ def setUp(self):
     def tearDown(self):
         super().tearDown()
 
-        del self.pipeline 
+        del self.pipeline
         gc.collect()
         torch.cuda.empty_cache()
 

tests/lora/test_lora_layers_sd3.py

@@ -33,7 +33,6 @@
     is_peft_available,
     nightly,
     numpy_cosine_similarity_distance,
-    print_tensor_test,
     require_big_gpu_with_torch_cuda,
     require_peft_backend,
     require_torch_gpu,
@@ -184,7 +183,6 @@ def test_sd3_img2img_lora(self):
 
         image = pipe(**inputs).images[0]
         image_slice = image[0, :10, :10]
-        print_tensor_test(image[0, :10, :10].flatten())
         expected_slice = np.array(
             [
                 0.47827148,

tests/pipelines/flux/test_pipeline_flux.py

@@ -9,6 +9,7 @@
 
 from diffusers import AutoencoderKL, FlowMatchEulerDiscreteScheduler, FluxPipeline, FluxTransformer2DModel
 from diffusers.utils.testing_utils import (
+    nightly,
     numpy_cosine_similarity_distance,
     require_big_gpu_with_torch_cuda,
     slow,
@@ -193,6 +194,7 @@ def test_fused_qkv_projections(self):
 
 
 @slow
+@nightly
 @require_big_gpu_with_torch_cuda
 @pytest.mark.big_gpu_with_torch_cuda
 class FluxPipelineSlowTests(unittest.TestCase):