using model_local_path

app.py

@@ -6,15 +6,28 @@ import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline
 import torch
 
-model_id = "ostris/Z-Image-De-Turbo"
+model_local_path = "./Z-Image-De-Turbo"  # Попробуйте сначала этот путь
+
+if not os.path.exists(model_local_path):
+    potential_paths = [
+        "/home/user/Z-Image-De-Turbo",
+        os.path.expanduser("~/.cache/huggingface/hub/models--ostris--Z-Image-De-Turbo"),
+    ]
+    for path in potential_paths:
+        if os.path.exists(path):
+            model_local_path = path
+            break
+
+print(f"Пытаюсь загрузить модель из: {model_local_path}")
 
 pipe = DiffusionPipeline.from_pretrained(
-    model_id,
+    os.path.join(model_local_path, "transformer"),
     torch_dtype=torch.bfloat16,
-    variant="bf16",
-    subfolder="transformer"
 )
 
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe.to(device)
+
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024