Fix import if PyTorch is not installed (#715)

* Fix import if PyTorch is not installed. * Style (blank line)

Fix import if PyTorch is not installed (#715)
* Fix import if PyTorch is not installed. * Style (blank line)
215bb408 · Pedro Cuenca · GitHub · 5ac1f61c · 215bb408
Unverified Commit 215bb408 authored Oct 04, 2022 by Pedro Cuenca Committed by GitHub Oct 04, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

src/diffusers/utils/__init__.py src/diffusers/utils/__init__.py +4 -1

No files found.
--- a/src/diffusers/utils/__init__.py
+++ b/src/diffusers/utils/__init__.py
@@ -37,7 +37,10 @@ from .import_utils import (
 )
 from .logging import get_logger
 from .outputs import BaseOutput
-from .testing_utils import floats_tensor, load_image, parse_flag_from_env, slow, torch_device
+
+
+if is_torch_available():
+    from .testing_utils import floats_tensor, load_image, parse_flag_from_env, slow, torch_device


 logger = get_logger(__name__)