From b6bea8f2c0a2037446951344823c3cf41ee5afdc Mon Sep 17 00:00:00 2001
From: hlyml <2024212846@stu.hit.edu.cn>
Date: Fri, 17 Oct 2025 11:03:26 +0800
Subject: [PATCH] Modify skip_cfg_above_sigma assignment logic

Adjust skip_cfg_above_sigma based on pixel count.
---
 nodes.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index 71f42b7..c62196c 100644
--- a/nodes.py
+++ b/nodes.py
@@ -105,7 +105,15 @@ class NovelAITXT2IMGPayload(BaseRequest):
         
         # 设置skip_cfg_above_sigma值
         if self.skip_cfg_above_sigma_enable:
-            self.skip_cfg_above_sigma = 59.04722600415217
+            pixel_count = self.width * self.height
+            if pixel_count == 1024 * 1024:
+                self.skip_cfg_above_sigma = 59.04722600415217
+            elif pixel_count == 1216 * 832:
+                self.skip_cfg_above_sigma = 58.0
+            elif pixel_count == 1408 * 704:
+                self.skip_cfg_above_sigma = 57.40995413696215
+            else:
+                self.skip_cfg_above_sigma = 58.0
         else:
             self.skip_cfg_above_sigma = None