change device to cpu

sd.py

@@ -47,7 +47,7 @@ from ldm.models.diffusion.ddim import DDIMSampler
 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=False):
+def load_model_from_config(config, ckpt, verbose=False, 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)
@@ -64,9 +64,7 @@ def load_model_from_config(config, ckpt, verbose=False, device='cuda', half_prec
         print(u)
 
     if half_precision:
-        model = model.half().to(device)
-    else:
-        model = model.to(device)
+        model = model.half()
     model.eval()
     return model
 
@@ -93,7 +91,8 @@ ckpt_path = hf_hub_download(repo_id="CompVis/stable-diffusion-v-1-4-original", f
 ckpt_config_path = "./stable-diffusion/configs/stable-diffusion/v1-inference.yaml"
 
 local_config = OmegaConf.load(f"{ckpt_config_path}")
-model = load_model_from_config(local_config, f"{ckpt_path}")
+
+model = load_model_from_config(local_config, f"{ckpt_path}" )
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 model = model.to(device)
 print('Model saved.')