Update model.py

model.py

@@ -140,8 +140,18 @@ class ModelHandler:
         print("Loading LoRA weights...")
         self.pipeline.load_lora_weights(Config.REPO_ID, weight_name=Config.LORA_FILENAME)
         
-        # --- FIX: REMOVED self.pipeline.fuse_lora() ---
-        # We do not fuse, so we can set the scale at runtime.
+        # --- NEW: Fuse LoRA at build time with fixed strength ---
+        print(f"Fusing LoRA with scale {Config.LORA_STRENGTH}...")
+        self.pipeline.fuse_lora(lora_scale=Config.LORA_STRENGTH)
+        print("  [OK] LoRA fused.")
+        
+        # --- NEW: Optimize UNet with torch.compile ---
+        try:
+            print("Compiling UNet with torch.compile...")
+            self.pipeline.unet = torch.compile(self.pipeline.unet, mode="reduce-overhead", fullgraph=True)
+            print("  [OK] UNet compiled.")
+        except Exception as e:
+            print(f"  [WARNING] torch.compile failed: {e}. Running without compilation.")
         
         # 6. Load Preprocessors
         print("Loading Preprocessors (Zoe, LineArt)...")
@@ -157,7 +167,7 @@ class ModelHandler:
             
         try:
             # Convert PIL to CV2
-            cv2_img = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR) 
+            cv2_img = cv2.cvtColor(np.array(image), cv2.COLOR_RGB_BGR) 
             faces = self.app.get(cv2_img)
             
             if len(faces) == 0: