Skip to content

Commit 8b275ce

Browse files
Support auto detecting some zsnr anime checkpoints.
1 parent 2a18e98 commit 8b275ce

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

comfy/supported_models.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -197,6 +197,8 @@ def model_type(self, state_dict, prefix=""):
197197
self.sampling_settings["sigma_min"] = float(state_dict["edm_vpred.sigma_min"].item())
198198
return model_base.ModelType.V_PREDICTION_EDM
199199
elif "v_pred" in state_dict:
200+
if "ztsnr" in state_dict: #Some zsnr anime checkpoints
201+
self.sampling_settings["zsnr"] = True
200202
return model_base.ModelType.V_PREDICTION
201203
else:
202204
return model_base.ModelType.EPS

0 commit comments

Comments
 (0)