Update modules/sd_models.py

modules/sd_models.py

@@ -414,7 +414,7 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer
     if devices.fp8:
         # prevent model to load state dict in fp8
         if torch.cuda.is_available():
-        model.half()
+            model.half()
        
 
     if not SkipWritingToConfig.skip:
@@ -484,7 +484,7 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer
         alphas_cumprod = model.alphas_cumprod
         model.alphas_cumprod = None
         if torch.cuda.is_available():
-        model.half()
+            model.half()
         model.alphas_cumprod = alphas_cumprod
         model.alphas_cumprod_original = alphas_cumprod
         model.first_stage_model = vae