QJerry commited on
Commit
be5f22f
·
verified ·
1 Parent(s): c1640c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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}/transformer",
 
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: