Spaces:

kael558
/

Interpolation

Runtime error

kael558 commited on Dec 17, 2022

Commit

9ad8b05

1 Parent(s): b5c0ab1

changed to cpu

Files changed (1) hide show

sd.py CHANGED Viewed

@@ -48,7 +48,7 @@ from ldm.models.diffusion.plms import PLMSSampler
 # 2. Set model download config
 def load_model_from_config(config, ckpt, verbose=False, device='cuda', half_precision=True):
-    map_location = "cuda" #@param ["cpu", "cuda"]
     print(f"Loading model from {ckpt}")
     pl_sd = torch.load(ckpt, map_location=map_location)
     if "global_step" in pl_sd:

 # 2. Set model download config
 def load_model_from_config(config, ckpt, verbose=False, device='cuda', half_precision=True):
+    map_location = "cuda" if torch.cuda.is_available() else "cpu"
     print(f"Loading model from {ckpt}")
     pl_sd = torch.load(ckpt, map_location=map_location)
     if "global_step" in pl_sd: