fix: loras rearrange when refreshed

Author: colinurbs

modules/interface.py

@@ -334,13 +334,16 @@ def apply_startup_settings():
         )
         connect_audio_events(a, settings)
 
-        def refresh_loras():
+        def refresh_loras(current_selected):
             if enumerate_lora_dir_fn:
                 new_lora_names = enumerate_lora_dir_fn()
-                return gr.update(choices=new_lora_names)
+                preserved = [name for name in (current_selected or []) if name in new_lora_names]
+                return gr.update(choices=new_lora_names, value=preserved)
             return gr.update()
 
-        g["refresh_loras_button"].click(fn=refresh_loras, outputs=[g["lora_selector"]])
+        g["refresh_loras_button"].click(
+            fn=refresh_loras, inputs=[g["lora_selector"]], outputs=[g["lora_selector"]]
+        )
 
         # General Connections
         def initial_gallery_load():

modules/ui/generate.py

@@ -273,7 +273,7 @@ def create_generate_ui(
                         latent_window_size = gr.Slider(
                             label="Latent Window Size",
                             minimum=1,
-                            maximum=33,
+                            maximum=60,
                             value=9,
                             step=1,
                             info="Change at your own risk, very experimental",
@@ -611,6 +611,16 @@ def process_with_queue_update(model_type_arg, *args):
             if is_ui_video_model and input_video_arg is not None
             else None
         )
+        # Realign LoRA names and weights to the stable slider order to prevent mis-mapping after refresh
+        # The slider components are created at UI build time and their order remains stable.
+        # After refreshing available LoRAs, choices can change, but we must keep lora_loaded_names (state)
+        # aligned with the slider input order to avoid mixing/misalignment of weights.
+        stable_slider_order = list(g["lora_sliders"].keys())
+        incoming_weight_by_name = dict(zip(stable_slider_order, lora_slider_values_tuple))
+        # Override the lora_names_states and weights passed to the backend to match the stable slider order
+        lora_names_states_arg = stable_slider_order
+        lora_slider_values_tuple = [incoming_weight_by_name.get(name, 1.0) for name in stable_slider_order]
+
         result = f["process_fn"](
             backend_model_type,
             input_data,

shared/Enums.py

@@ -4,6 +4,8 @@
 
 from shared.StrEnum import StrEnum
 
+logger = logging.getLogger(__name__)
+
 
 class QuantizationFormat(StrEnum):
     normal_float_4bit = auto()
@@ -12,7 +14,6 @@ class QuantizationFormat(StrEnum):
     NONE = brain_floating_point_16bit
     DEFAULT = NONE
 
-    __logger = logging.getLogger(__name__)
 
     @staticmethod
     def supported_values() -> list[str]:
@@ -22,19 +23,19 @@ def supported_values() -> list[str]:
     @staticmethod
     def safe_parse(value: "str | QuantizationFormat") -> "QuantizationFormat":
         if find_spec("bitsandbytes") is None:
-            QuantizationFormat.__logger.error(
+            logger.error(
                 "bitsandbytes not found, defaulting to no quantization. https://docs.framepackstudio.com/help"
             )
             # If bitsandbytes is not installed, we can not quantize, so return NONE
             return QuantizationFormat.NONE
 
-        QuantizationFormat.__logger.debug(f"QuantizationFormat: {value}")
+        logger.debug(f"QuantizationFormat: {value}")
         if isinstance(value, QuantizationFormat):
             return value
         try:
             return QuantizationFormat(value)
         except ValueError:
-            QuantizationFormat.__logger.exception(
+            logger.exception(
                 f"Invalid QuantizationFormat value: {value}, defaulting to {QuantizationFormat.DEFAULT}."
             )
             return QuantizationFormat.DEFAULT
@@ -45,7 +46,6 @@ class LoraLoader(StrEnum):
     LORA_READY = "lora_ready"
     DEFAULT = LORA_READY
 
-    __logger = logging.getLogger(__name__)
 
     @staticmethod
     def supported_values() -> list[str]:
@@ -59,7 +59,7 @@ def safe_parse(value: "str | LoraLoader") -> "LoraLoader":
         try:
             return LoraLoader(value)
         except ValueError:
-            LoraLoader.__logger.exception(
+            logger.exception(
                 f"Invalid LoraLoader value: {value}, defaulting to {LoraLoader.DEFAULT}."
             )
             return LoraLoader.DEFAULT

studio.py

@@ -265,7 +265,7 @@ def enumerate_lora_dir() -> list[str]:
                         )
                         lora_name = str(PurePath(lora_relative_path).with_suffix(""))
                         found_files.append(lora_name)
-            found_files.sort()
+            found_files.sort(key=lambda s: s.casefold())
             print(f"Found LoRAs: {len(found_files)}")
             # Temp solution for only 1 lora
             if len(found_files) == 1: