Modify skip_cfg_above_sigma assignment logic

Adjust skip_cfg_above_sigma based on pixel count.
2025-10-17 11:03:26 +08:00
parent 24b92984c3
commit b6bea8f2c0
1 changed files with 9 additions and 1 deletions
@@ -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