手动补充scaling

.gitignore

@@ -0,0 +1,4 @@
+.vscode
+.DS_Store
+__pycache__
+checkpoints

OminiControl/src/flux/lora_controller.py

@@ -7,6 +7,13 @@ class enable_lora:
         self.activated: bool = activated
         if activated:
             return
+        
+        for lora_module in lora_modules:
+            for active_adapter in lora_module.active_adapters:
+                scaling = lora_module.scaling
+                if active_adapter not in scaling:
+                    scaling[active_adapter] = 1.0
+
         self.lora_modules: List[BaseTunerLayer] = [
             each for each in lora_modules if isinstance(each, BaseTunerLayer)
         ]

ominicontrol.py

@@ -3,7 +3,6 @@ from diffusers.pipelines import FluxPipeline
 from OminiControl.src.flux.condition import Condition
 from PIL import Image
 import random
-import math
 
 from OminiControl.src.flux.generate import generate, seed_everything
 
@@ -43,7 +42,8 @@ pipe.load_lora_weights(
 )
 # ref: https://civitai.com/models/824739/flux-3d-animation-style-lora
 pipe.load_lora_weights(
-    "./lora/3d_animation.safetensors",
+    "./lora",
+    weight_name="3d_animation.safetensors",
     adapter_name="3d_animation",
 )