Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -135,7 +135,8 @@ def load_models(model_path, enable_compile=False, attention_backend="native"):
|
|
| 135 |
|
| 136 |
if not os.path.exists(model_path):
|
| 137 |
transformer = ZImageTransformer2DModel.from_pretrained(
|
| 138 |
-
f"{model_path}
|
|
|
|
| 139 |
use_auth_token=use_auth_token
|
| 140 |
).to("cuda", torch.bfloat16)
|
| 141 |
else:
|
|
|
|
| 135 |
|
| 136 |
if not os.path.exists(model_path):
|
| 137 |
transformer = ZImageTransformer2DModel.from_pretrained(
|
| 138 |
+
f"{model_path}",
|
| 139 |
+
subfolder="transformer",
|
| 140 |
use_auth_token=use_auth_token
|
| 141 |
).to("cuda", torch.bfloat16)
|
| 142 |
else:
|