revert

OminiControl/src/flux/lora_controller.py

@@ -7,12 +7,6 @@ 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)

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-import spaces
 from ominicontrol import generate_image
 import os
 
@@ -195,6 +194,7 @@ def infer(
 
 
 if USE_ZERO_GPU:
+    import spaces
     infer = spaces.GPU(infer)
 
 if __name__ == "__main__":