Update modules/sd_models.py

modules/sd_models.py

@@ -874,7 +874,7 @@ def load_model(checkpoint_info=None, already_loaded_state_dict=None):
 
     if not torch.cuda.is_available():
         print("Running on CPU: forcing model to float32")
-        sd_model.to(torch.float32)
+        sd_model = sd_model.float()
 
     with torch.no_grad():
         sd_model.cond_stage_model_empty_prompt = get_empty_cond(sd_model)