Update app.py

app.py

@@ -13,6 +13,9 @@ import random
 import re
 from PIL import ImageFilter  # Für GaussianBlur wird nur für SAM benötigt!
 
+
+os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"  # 2x schnellerer Download
+
 # === OPTIMIERTE EINSTELLUNGEN ===
 device = "cuda" if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.float16 if device == "cuda" else torch.float32
@@ -576,7 +579,9 @@ def load_img2img(keep_environment=False):
                     controlnet=[controlnet_depth, controlnet_canny],  # Depth zuerst!
                     torch_dtype=torch_dtype,
                     safety_checker=None,
-                    requires_safety_checker=False
+                    requires_safety_checker=False,
+                    cache_dir="/tmp/models",
+                    use_safetensors=True
                 ).to(device)
                 
                 # Scheduler konfigurieren
@@ -619,7 +624,9 @@ def load_img2img(keep_environment=False):
                     controlnet=[controlnet_openpose, controlnet_canny],  # OpenPose zuerst!
                     torch_dtype=torch_dtype,
                     safety_checker=None,
-                    requires_safety_checker=False
+                    requires_safety_checker=False,
+                    cache_dir="/tmp/models",
+                    use_safetensors=True
                 ).to(device)
                 
                 # Scheduler konfigurieren