diff --git a/comfy/supported_models.py b/comfy/supported_models.py index 9931f4c5..75ddaee5 100644 --- 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