cpu

VADER-VideoCrafter/scripts/main/train_t2v_lora.py

@@ -710,7 +710,7 @@ def run_training(args, model, **kwargs):
         mixed_precision=args.mixed_precision,
         project_dir=args.project_dir,
         device_placement=True,
-        cpu=False
+        cpu=True
     )
     output_dir = args.project_dir
 

app.py

@@ -22,7 +22,7 @@ examples = [
 
 model = setup_model()
 
-@spaces.GPU(duration=120)
+# @spaces.GPU(duration=180)
 def gradio_main_fn(prompt, lora_model, lora_rank, seed, height, width, unconditional_guidance_scale, ddim_steps, ddim_eta,
                    frames, savefps):
     global model