Update ldm/inference_base.py

ldm/inference_base.py

@@ -175,8 +175,10 @@ def get_sd_models(opt):
     build stable diffusion model, sampler
     """
     # SD
+    device = 'cuda' if torch.cuda.is_available() else 'cpu'
     config = OmegaConf.load(f"{opt.config}")
     model = load_model_from_config(config, opt.sd_ckpt, opt.vae_ckpt)
+    model = model.half()
     sd_model = model.to(opt.device)
 
     # sampler