BryanW commited on
Commit
f955714
·
verified ·
1 Parent(s): 6ab28e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,11 +12,11 @@ import numpy as np
12
 
13
 
14
  def load_models(model_path="MeissonFlow/Meissonic",
15
- transformer_path="MeissonFlow/Muddit/1024",
16
  device="cuda"):
17
  model = SymmetricTransformer2DModel.from_pretrained(
18
  transformer_path,
19
- subfolder="transformer",
20
  )
21
  vq_model = VQModel.from_pretrained(model_path, subfolder="vqvae")
22
  text_encoder = CLIPTextModelWithProjection.from_pretrained(model_path, subfolder="text_encoder")
 
12
 
13
 
14
  def load_models(model_path="MeissonFlow/Meissonic",
15
+ transformer_path="MeissonFlow/Muddit",
16
  device="cuda"):
17
  model = SymmetricTransformer2DModel.from_pretrained(
18
  transformer_path,
19
+ subfolder="1024/transformer",
20
  )
21
  vq_model = VQModel.from_pretrained(model_path, subfolder="vqvae")
22
  text_encoder = CLIPTextModelWithProjection.from_pretrained(model_path, subfolder="text_encoder")