Update modules/sd_models.py

modules/sd_models.py

@@ -415,6 +415,8 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer
         # prevent model to load state dict in fp8
         if torch.cuda.is_available():
             model.half()
+        else:
+            print("CUDA not available — skipping model.half()")
        
 
     if not SkipWritingToConfig.skip:
@@ -485,6 +487,8 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer
         model.alphas_cumprod = None
         if torch.cuda.is_available():
             model.half()
+        else:
+            print("CUDA not available — skipping model.half()")
         model.alphas_cumprod = alphas_cumprod
         model.alphas_cumprod_original = alphas_cumprod
         model.first_stage_model = vae