Update generator.py

generator.py

@@ -319,22 +319,32 @@ class RetroArtConverter:
                 print(f"  [WARNING] Face detection failed: {e}")
         
         # Unfuse and reload LORA with new scale (like exampleapp.py)
-        if hasattr(self.pipe, 'unfuse_lora'):
-            try:
-                self.pipe.unfuse_lora()
-                self.pipe.unload_lora_weights()
-                print("  [OK] Unfused previous LORA")
-            except Exception as e:
-                print(f"  [INFO] No previous LORA to unfuse: {e}")
+        #if hasattr(self.pipe, 'unfuse_lora'):
+        #    try:
+        #        self.pipe.unfuse_lora()
+        #        self.pipe.unload_lora_weights()
+        #        print("  [OK] Unfused previous LORA")
+        #    except Exception as e:
+        #        print(f"  [INFO] No previous LORA to unfuse: {e}")
         
         # Load and fuse LORA at the requested scale
-        if self.models_loaded['lora'] and self.models_loaded.get('lora_path'):
+        #if self.models_loaded['lora'] and self.models_loaded.get('lora_path'):
+        #    try:
+        #        self.pipe.load_lora_weights(self.models_loaded['lora_path'])
+        #        self.pipe.fuse_lora(lora_scale=lora_scale)
+        #        print(f"  [OK] LORA fused at scale: {lora_scale}")
+        #    except Exception as e:
+        #        print(f"  [WARNING] Could not fuse LORA: {e}")
+       
+        # --- CORRECTED BLOCK ---
+        # Set LORA scale using set_adapters (matches models.py)
+        if hasattr(self.pipe, 'set_adapters') and self.models_loaded['lora']:
             try:
-                self.pipe.load_lora_weights(self.models_loaded['lora_path'])
-                self.pipe.fuse_lora(lora_scale=lora_scale)
-                print(f"  [OK] LORA fused at scale: {lora_scale}")
+                self.pipe.set_adapters(["retroart"], adapter_weights=[lora_scale])
+                print(f"LORA scale: {lora_scale}")
             except Exception as e:
-                print(f"  [WARNING] Could not fuse LORA: {e}")
+                print(f"Could not set LORA scale: {e}")
+        # --- END OF BLOCK ---
         
         # Setup generator with seed control
         if seed == -1:
@@ -349,6 +359,15 @@ class RetroArtConverter:
         # Use Compel for prompt encoding (like exampleapp.py - simpler)
         if self.use_compel and self.compel is not None:
             print("Encoding prompts with Compel...")
+
+            # --- FIX: Add the LORA trigger word ---
+            # Ensure trigger word is present and avoid duplicates
+            if TRIGGER_WORD not in prompt:
+                # Prepend the trigger word for highest impact
+                prompt = f"{TRIGGER_WORD}, {prompt}"
+            print(f"  Using final prompt: {prompt}")
+            # --- End Fix ---
+
             conditioning, pooled = self.compel(prompt)
             negative_conditioning, negative_pooled = self.compel(negative_prompt)
             print("  [OK] Prompts encoded")