Spaces:

MeissonFlow
/

muddit

Running on Zero

BryanW commited on 26 days ago

Commit

a9f1a7f

verified ·

1 Parent(s): d8a901d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ def load_models(model_path="MeissonFlow/Meissonic",
     model = SymmetricTransformer2DModel.from_pretrained(
         transformer_path,
         subfolder="1024/transformer",
-        torch_dtype=dtype)
     )
     vq_model = VQModel.from_pretrained(model_path, subfolder="vqvae",torch_dtype=dtype)
     text_encoder = CLIPTextModelWithProjection.from_pretrained(model_path, subfolder="text_encoder",torch_dtype=dtype)

     model = SymmetricTransformer2DModel.from_pretrained(
         transformer_path,
         subfolder="1024/transformer",
+        torch_dtype=dtype
     )
     vq_model = VQModel.from_pretrained(model_path, subfolder="vqvae",torch_dtype=dtype)
     text_encoder = CLIPTextModelWithProjection.from_pretrained(model_path, subfolder="text_encoder",torch_dtype=dtype)