debug

app.py

@@ -26,6 +26,14 @@ from videox_fun.data.dataset_image_video import derive_ground_object_from_instru
 from videox_fun.utils.lora_utils import merge_lora, unmerge_lora
 from videox_fun.utils.utils import save_videos_grid, timer
 
+global_controller = None
+
+@spaces.GPU(duration=300)
+@timer
+def generate_wrapper(*args):
+    global global_controller
+    return global_controller.generate(*args)
+
 # Redefine create_height_width to remove Chinese and specific defaults if needed, 
 # although we will mostly ignore sliders if we use input resolution.
 # We will create a custom version here to avoid modifying the library file if possible,
@@ -150,7 +158,6 @@ def preload_models(controller, default_model_path, default_lora_name, acc_lora_p
         torch.cuda.empty_cache()
 
 class VideoCoF_Controller(Wan_Controller):
-    @spaces.GPU(duration=300)
     @timer
     def generate(
         self,
@@ -362,6 +369,8 @@ def ui(GPU_memory_mode, scheduler_dict, config_path, compile_dit, weight_dtype):
         config_path=config_path, compile_dit=compile_dit,
         weight_dtype=weight_dtype
     )
+    global global_controller
+    global_controller = controller
 
     with gr.Blocks() as demo:
         gr.Markdown("# VideoCoF Demo")
@@ -472,7 +481,7 @@ def ui(GPU_memory_mode, scheduler_dict, config_path, compile_dit, weight_dtype):
 
         # Event handlers
         generate_button.click(
-            fn=controller.generate,
+            fn=generate_wrapper,
             inputs=[
                 diffusion_transformer_dropdown,
                 base_model_dropdown,