Spaces:

jjz5463
/

Diary-AI-Video

Paused

jjz5463 commited on Oct 5, 2024

Commit

ed65d52

1 Parent(s): b8e08a3

quick inference test

Files changed (1) hide show

baseline_utils.py CHANGED Viewed

@@ -151,7 +151,7 @@ def generate_video(scene_list, fps=24):  # Lower fps
     video = pipe(
         prompt=scene_list,
         num_videos_per_prompt=1,
-        num_inference_steps=40,
         num_frames=fps,
         guidance_scale=6,
         generator=torch.Generator(device=device).manual_seed(42),

     video = pipe(
         prompt=scene_list,
         num_videos_per_prompt=1,
+        num_inference_steps=1,
         num_frames=fps,
         guidance_scale=6,
         generator=torch.Generator(device=device).manual_seed(42),