Fix some fp8 scaled checkpoints no longer working. (#13239)
parent
94ab69f609
commit
4504ee718a
|
|
@ -1724,6 +1724,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)
|
||||
|
||||
custom_operations = model_options.get("custom_operations", None)
|
||||
if custom_operations is None:
|
||||
|
|
|
|||
Loading…
Reference in New Issue