modified: gradio_app.py

gradio_app.py

@@ -100,7 +100,16 @@ def load_target_model(selected_model):
         t5xxl.eval()
         ae = flux_utils.load_ae(AE_PATH, torch.bfloat16, "cpu", disable_mmap=False)
         logger.info("Models loaded successfully.")
-        return model, [clip_l, t5xxl], ae
+
+        # Load LoRA weights
+        multiplier = 1.0
+        weights_sd = load_file(LORA_WEIGHTS_PATH)
+        lora_model, _ = lora_flux.create_network_from_weights(multiplier, None, ae, [clip_l, t5xxl], model, weights_sd, True)
+        lora_model.apply_to([clip_l, t5xxl], model)
+        info = lora_model.load_state_dict(weights_sd, strict=True)
+        logger.info(f"Loaded LoRA weights from {LORA_WEIGHTS_PATH}: {info}")
+        lora_model.eval()
+        
     except Exception as e:
         logger.error(f"Error loading models: {e}")
         raise