Update modules/sd_models.py

modules/sd_models.py

@@ -872,6 +872,10 @@ def load_model(checkpoint_info=None, already_loaded_state_dict=None):
 
     timer.record("scripts callbacks")
 
+    if not torch.cuda.is_available():
+        print("Running on CPU: forcing model to float32")
+        sd_model.to(torch.float32)
+
     with torch.no_grad():
         sd_model.cond_stage_model_empty_prompt = get_empty_cond(sd_model)