Fix some fp8 scaled checkpoints no longer working.

2026-04-05 04:09:08 +00:00 · 2026-03-31 11:21:49 -04:00
1 changed files with 2 additions and 0 deletions
--- a/comfy/sd.py
+++ b/comfy/sd.py
@@ -1745,6 +1745,8 @@ def load_diffusion_model_state_dict(sd, model_options={}, metadata=None, disable
    temp_sd = comfy.utils.state_dict_prefix_replace(sd, {diffusion_model_prefix: ""}, filter_keys=True)
    if len(temp_sd) > 0:
        sd = temp_sd
+        if custom_operations is None:
+            sd, metadata = comfy.utils.convert_old_quants(sd, "", metadata=metadata)

    parameters = comfy.utils.calculate_parameters(sd)
    weight_dtype = comfy.utils.weight_dtype(sd)