From 8b275ce5be29ff7d847c3c4c2f3fea1faa68e07b Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Mon, 11 Nov 2024 05:25:16 -0500
Subject: [PATCH] Support auto detecting some zsnr anime checkpoints.

---
 comfy/supported_models.py | 2 ++
 1 file changed, 2 insertions(+)

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