Implement noise augmentation for SD 4X upscale model.

2025-09-12 20:48:22 +00:00 · 2024-01-03 14:27:11 -05:00
parent ef4f6037cb
commit 8c6493578b
6 changed files with 33 additions and 14 deletions
--- a/comfy/supported_models.py
+++ b/comfy/supported_models.py
@@ -290,6 +290,7 @@ class SD_X4Upscaler(SD20):

    unet_extra_config = {
        "disable_self_attentions": [True, True, True, False],
+        "num_classes": 1000,
        "num_heads": 8,
        "num_head_channels": -1,
    }