Spaces:

MeissonFlow
/

muddit

Runtime error

BryanW commited on Jul 3

Commit

55950a3

verified ·

1 Parent(s): a9f1a7f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,6 +14,7 @@ import spaces
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 dtype = torch.bfloat16
 def load_models(model_path="MeissonFlow/Meissonic",
             transformer_path="MeissonFlow/Muddit"):
     model = SymmetricTransformer2DModel.from_pretrained(

 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 dtype = torch.bfloat16
+@spaces.GPU
 def load_models(model_path="MeissonFlow/Meissonic",
             transformer_path="MeissonFlow/Muddit"):
     model = SymmetricTransformer2DModel.from_pretrained(