diff --git a/examples/inference/basic/basic_hy15.py b/examples/inference/basic/basic_hy15.py index eac65afe9..e0b782237 100644 --- a/examples/inference/basic/basic_hy15.py +++ b/examples/inference/basic/basic_hy15.py @@ -2,18 +2,18 @@ from fastvideo import VideoGenerator import json # from fastvideo.configs.sample import SamplingParam -OUTPUT_PATH = "video_samples_hy15_step_distilled" +OUTPUT_PATH = "video_samples_hy15" def main(): # FastVideo will automatically use the optimal default arguments for the # model. # If a local path is provided, FastVideo will make a best effort # attempt to identify the optimal arguments. generator = VideoGenerator.from_pretrained( - "hunyuanvideo-community/HunyuanVideo-1.5-Diffusers-480p_i2v_step_distilled", + "hunyuanvideo-community/HunyuanVideo-1.5-Diffusers-480p_t2v", # FastVideo will automatically handle distributed setup num_gpus=1, use_fsdp_inference=False, # set to True if GPU is out of memory - dit_cpu_offload=False, + dit_cpu_offload=True, vae_cpu_offload=True, text_encoder_cpu_offload=True, pin_cpu_memory=True, # set to false if low CPU RAM or hit obscure "CUDA error: Invalid argument" @@ -26,7 +26,7 @@ def main(): "natural light filtering through the petals. Mid-shot, warm and cheerful tones." ) - video = generator.generate_video(prompt, output_path=OUTPUT_PATH, save_video=True, negative_prompt="") + video = generator.generate_video(prompt, output_path=OUTPUT_PATH, save_video=True, negative_prompt="", num_frames=81, fps=16) prompt2 = ( "A majestic lion strides across the golden savanna, its powerful frame " @@ -35,8 +35,8 @@ def main(): "embodying the raw energy of the wild. Low angle, steady tracking shot, " "cinematic.") - video2 = generator.generate_video(prompt2, output_path=OUTPUT_PATH, save_video=True, negative_prompt="") + video2 = generator.generate_video(prompt2, output_path=OUTPUT_PATH, save_video=True, negative_prompt="", num_frames=81, fps=16) if __name__ == "__main__": - main() + main() \ No newline at end of file