Spaces:

Tasty-Rice
/

Magic_on_paper

Runtime error

TastyRice commited on May 10, 2024

Commit

cb95979

verified ·

1 Parent(s): 7aa3d31

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -37,7 +37,7 @@ torch.backends.cudnn.deterministic = True
 torch.backends.cudnn.benchmark = False
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
-pipeline.to(device)
 def load_pipeline(model_name):
     vae = AutoencoderKL.from_pretrained(

 torch.backends.cudnn.benchmark = False
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
+MODEL.to(device)
 def load_pipeline(model_name):
     vae = AutoencoderKL.from_pretrained(