Fix import from diffusers latest release (#1380)

huggingface · Sep 14, 2023 · 656e7af · 656e7af
1 parent afd2b5a
commit 656e7af
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/onnxruntime/test_stable_diffusion_pipeline.py b/tests/onnxruntime/test_stable_diffusion_pipeline.py
@@ -25,7 +25,8 @@
     StableDiffusionPipeline,
     StableDiffusionXLPipeline,
 )
-from diffusers.utils import floats_tensor, load_image
+from diffusers.utils import load_image
+from diffusers.utils.testing_utils import floats_tensor
 from parameterized import parameterized
 from transformers.testing_utils import require_torch_gpu
 from utils_onnxruntime_tests import MODEL_NAMES, SEED, ORTModelTestMixin