duration

app.py

@@ -3,22 +3,15 @@ import os
 os.environ["GRADIO_TEMP_DIR"] = os.path.join(os.getcwd(), ".tmp_outputs")
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
 
-import logging
 import uuid
 
 import GPUtil
 import gradio as gr
 import psutil
 import spaces
-import torch
 
 from videosys import CogVideoXConfig, CogVideoXPABConfig, VideoSysEngine
 
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-dtype = torch.float16
-
 
 def load_model(enable_video_sys=False, pab_threshold=[100, 850], pab_range=2):
     pab_config = CogVideoXPABConfig(spatial_threshold=pab_threshold, spatial_range=pab_range)
@@ -57,14 +50,14 @@ def get_server_status():
     return {"cpu": f"{cpu_percent}%", "memory": f"{memory.percent}%", "disk": f"{disk.percent}%", "gpu": gpu_info}
 
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=240)
 def generate_vanilla(prompt, num_inference_steps, guidance_scale, progress=gr.Progress(track_tqdm=True)):
     engine = load_model()
     video_path = generate(engine, prompt, num_inference_steps, guidance_scale)
     return video_path
 
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=240)
 def generate_vs(
     prompt,
     num_inference_steps,