Add CLIP skip spin box to Common Options tab.

(partial fix for #16)

backend/config.py

@@ -28,6 +28,10 @@ class GenerationOptions(BaseModel):
     """Model to use for generation."""
     sd_vae: str = "Automatic"
     """VAE to use for generation."""
+
+    clip_skip: int = 1
+    """CLIP layers to skip during generation."""
+
     script: str = "None"
     """Which script to use."""
     script_args: list = Field(default_factory=list)

backend/utils.py

@@ -89,6 +89,9 @@ def prepare_backend(opt: BaseModel):
         shared.opts.sd_vae = opt.sd_vae
         modules.sd_vae.reload_vae_weights()
 
+    if hasattr(opt, "clip_skip"):
+        shared.opts.CLIP_stop_at_last_layers = opt.clip_skip
+
     if hasattr(opt, "upscaler_name"):
         shared.opts.upscaler_for_img2img = opt.upscaler_name
 

frontends/krita/krita_diff/client.py

@@ -221,6 +221,7 @@ def common_params(self, has_selection):
         params = dict(
             sd_model=self.cfg("sd_model", str),
             sd_vae=self.cfg("sd_vae", str),
+            clip_skip=self.cfg("clip_skip", int),
             batch_count=self.cfg("sd_batch_count", int),
             batch_size=self.cfg("sd_batch_size", int),
             base_size=self.cfg("sd_base_size", int),

frontends/krita/krita_diff/defaults.py

@@ -65,6 +65,7 @@ class Defaults:
     sd_model: str = "model.ckpt"
     sd_vae_list: List[str] = field(default_factory=lambda: [ERROR_MSG])
     sd_vae: str = "Automatic"
+    clip_skip: int = 1
     sd_batch_size: int = 1
     sd_batch_count: int = 1
     sd_base_size: int = 512

frontends/krita/krita_diff/pages/common.py

@@ -26,6 +26,11 @@ def __init__(self, *args, **kwargs):
             script.cfg, "sd_vae_list", "sd_vae", label="VAE:"
         )
 
+        # Clip skip
+        self.clip_skip_layout = QSpinBoxLayout(
+            script.cfg, "clip_skip", label="Clip skip:", min=1, max=12, step=1
+        )
+
         # batch size & count
         self.batch_count_layout = QSpinBoxLayout(
             script.cfg, "sd_batch_count", label="Batch count:", min=1, max=9999, step=1
@@ -89,6 +94,7 @@ def __init__(self, *args, **kwargs):
         layout.addLayout(checkboxes_layout)
         layout.addLayout(self.sd_model_layout)
         layout.addLayout(self.sd_vae_layout)
+        layout.addLayout(self.clip_skip_layout)
         layout.addLayout(batch_layout)
         layout.addLayout(size_layout)
         layout.addWidget(self.interrupt_btn)
@@ -99,6 +105,7 @@ def __init__(self, *args, **kwargs):
     def cfg_init(self):
         self.sd_model_layout.cfg_init()
         self.sd_vae_layout.cfg_init()
+        self.clip_skip_layout.cfg_init()
         self.batch_count_layout.cfg_init()
         self.batch_size_layout.cfg_init()
         self.base_size_layout.cfg_init()
@@ -114,6 +121,7 @@ def cfg_init(self):
     def cfg_connect(self):
         self.sd_model_layout.cfg_connect()
         self.sd_vae_layout.cfg_connect()
+        self.clip_skip_layout.cfg_connect()
         self.batch_count_layout.cfg_connect()
         self.batch_size_layout.cfg_connect()
         self.base_size_layout.cfg_connect()