Update app.py

app.py

@@ -19,8 +19,8 @@ from ldm.models.diffusion.plms import PLMSSampler
 
 def load_model_from_config(config, ckpt, verbose=False):
     print(f"Loading model from {ckpt}")
-    # pl_sd = torch.load(ckpt, map_location="cpu")
-    pl_sd = torch.load(ckpt)#, map_location="cpu")
+    pl_sd = torch.load(ckpt, map_location="cpu")
+    # pl_sd = torch.load(ckpt)#, map_location="cpu")
     sd = pl_sd["state_dict"]
     model = instantiate_from_config(config.model)
     m, u = model.load_state_dict(sd, strict=False)