fix imports

Author: linoytsaban

tests/pipelines/sana/test_sana_sprint_img2img.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import inspect
+import random
 import unittest
 
 import numpy as np
@@ -22,6 +23,7 @@
 from diffusers import AutoencoderDC, SanaSprintImg2ImgPipeline, SanaTransformer2DModel, SCMScheduler
 from diffusers.utils.testing_utils import (
     enable_full_determinism,
+    floats_tensor,
     torch_device,
 )
 
@@ -38,7 +40,7 @@
 
 class SanaSprintImg2ImgPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
     pipeline_class = SanaSprintImg2ImgPipeline
-    params = TEXT_TO_IMAGE_PARAMS - {"cross_attention_kwargs", "negative_prompt", "negative_prompt_embeds"}
+    params = TEXT_GUIDED_IMAGE_VARIATION_PARAMS - {"cross_attention_kwargs", "negative_prompt", "negative_prompt_embeds"}
     batch_params = TEXT_GUIDED_IMAGE_VARIATION_BATCH_PARAMS - {"negative_prompt"}
     image_params = IMAGE_TO_IMAGE_IMAGE_PARAMS - {"negative_prompt"}
     image_latents_params = IMAGE_TO_IMAGE_IMAGE_PARAMS