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

Update app.py

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