dependency

Author: hlky

src/diffusers/utils/remote_utils.py

@@ -17,16 +17,15 @@
     if is_safetensors_available():
         import safetensors
 
-from PIL import Image
-
+    DTYPE_MAP = {
+        "float16": torch.float16,
+        "float32": torch.float32,
+        "bfloat16": torch.bfloat16,
+        "uint8": torch.uint8,
+    }
 
-DTYPE_MAP = {
-    "float16": torch.float16,
-    "float32": torch.float32,
-    "bfloat16": torch.bfloat16,
-    "uint8": torch.uint8,
-}
 
+from PIL import Image
 
 def remote_decode(
     endpoint: str,