subfolder param

app.py

@@ -6,27 +6,7 @@ import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline
 import torch
 
-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(
-    os.path.join(model_local_path, "transformer"),
-    torch_dtype=torch.bfloat16,
-)
-
-device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe.to(device)
+pipe = DiffusionPipeline.from_pretrained("ostris/Z-Image-De-Turbo", dtype=torch.bfloat16, subfolder='transformer', device_map="cuda")
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024