Fix UI overwriting previously set preferred checkpoint resolution

Acly · Jan 5, 2024 · 110d39f · 110d39f
1 parent e2d62fd
commit 110d39f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/ai_diffusion/settings.py b/ai_diffusion/settings.py
@@ -160,8 +160,8 @@ class Settings(QObject):
     _resolution_multiplier = Setting(
         "Resolution Multiplier",
         1.0,
-        "Scaling factor for image generation. Values below 1.0 improve performance when working"
-        " on high resolution canvas.",
+        "Scaling factor for generation. Values below 1.0 improve performance for high resolution"
+        " canvas.",
     )
 
     max_pixel_count: int

diff --git a/ai_diffusion/ui/settings.py b/ai_diffusion/ui/settings.py
@@ -965,10 +965,10 @@ def _set_checkpoint_warning(self):
                 self._checkpoint_warning.setVisible(True)
 
     def _toggle_preferred_resolution(self, checked: bool):
-        if checked:
+        if checked and self._resolution_spin.value == 0:
             sd_ver = resolve_sd_version(self.current_style, root.connection.client_if_connected)
             self._resolution_spin.value = 640 if sd_ver is SDVersion.sd15 else 1024
-        else:
+        elif not checked and self._resolution_spin.value > 0:
             self._resolution_spin.value = 0
 
     def _toggle_checkpoint_advanced(self, checked: bool):