Support auto detecting some zsnr anime checkpoints.

2024-11-11 05:25:16 -05:00 · 2024-11-11 05:25:16 -05:00 · 8b275ce5be
parent 2a18e98ccf
commit 8b275ce5be
1 changed files with 2 additions and 0 deletions
--- a/comfy/supported_models.py
+++ b/comfy/supported_models.py
@ -197,6 +197,8 @@ class SDXL(supported_models_base.BASE):
                self.sampling_settings["sigma_min"] = float(state_dict["edm_vpred.sigma_min"].item())
            return model_base.ModelType.V_PREDICTION_EDM
        elif "v_pred" in state_dict:
+            if "ztsnr" in state_dict: #Some zsnr anime checkpoints
+                self.sampling_settings["zsnr"] = True
            return model_base.ModelType.V_PREDICTION
        else:
            return model_base.ModelType.EPS