Fix not being able to switch back from edit mode (linked edit style) #2282

Author: Acly

ai_diffusion/ui/generation.py

@@ -12,6 +12,7 @@
 
 from ..properties import Binding, Bind, bind, bind_combo, bind_toggle
 from ..image import Bounds, Extent, Image
+from ..client import resolve_arch
 from ..jobs import Job, JobQueue, JobState, JobKind, JobParams
 from ..model import Model, InpaintContext, RootRegion, ProgressKind, Workspace
 from ..style import Styles
@@ -944,7 +945,8 @@ def update_generate_options(self):
         has_active_region = regions.is_linked(self.model.layers.active)
         is_region_only = has_regions and has_active_region and self.model.region_only
         is_edit = self.model.is_editing
-        can_switch_edit = self.model.can_edit and not arch.is_edit
+        base_arch = resolve_arch(self.model.style, root.connection.client_if_connected)
+        can_switch_edit = self.model.can_edit and not base_arch.is_edit
         self.region_mask_button.setVisible(has_regions)
         self.region_mask_button.setEnabled(has_active_region)
         self.region_mask_button.setIcon(_region_mask_button_icons[is_region_only])