Set model to fp16 before loading the state dict to lower ram bump.

im-stuff · Jun 14, 2023 · 6b77458 · 6b77458
1 parent 0c7cad4
commit 6b77458
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/comfy/sd.py b/comfy/sd.py
@@ -1155,9 +1155,9 @@ class WeightsLoader(torch.nn.Module):
     else:
         model = model_base.BaseModel(unet_config, v_prediction=v_prediction)
 
-    model = load_model_weights(model, sd, verbose=False, load_state_dict_to=load_state_dict_to)
-
     if fp16:
         model = model.half()
 
+    model = load_model_weights(model, sd, verbose=False, load_state_dict_to=load_state_dict_to)
+
     return (ModelPatcher(model), clip, vae, clipvision)