Update app.py

app.py

@@ -110,6 +110,9 @@ if "variance_type" in pipe.scheduler.config:
 pipe.scheduler = CogVideoXDPMScheduler.from_config(pipe.scheduler.config, **scheduler_args)
 pipe.to(device)
 
+pipe.enable_model_cpu_offload()
+pipe.enable_sequential_cpu_offload()
+
 os.makedirs("./output", exist_ok=True)
 os.makedirs("./gradio_tmp", exist_ok=True)