From c8a118cefd4d3932526c08485595a6aa39d4f439 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Fri, 29 Nov 2024 16:23:14 +0530 Subject: [PATCH] fix format specifier. --- training/mochi-1/text_to_video_lora.py | 22 ++++++++++------------ 1 file changed, 10 insertions(+), 12 deletions(-) diff --git a/training/mochi-1/text_to_video_lora.py b/training/mochi-1/text_to_video_lora.py index faa6255..fb0d69a 100644 --- a/training/mochi-1/text_to_video_lora.py +++ b/training/mochi-1/text_to_video_lora.py @@ -110,18 +110,16 @@ pipe = MochiPipeline.from_pretrained("genmo/mochi-1-preview") pipe.load_lora_weights("CHANGE_ME") pipe.enable_model_cpu_offload() -pipeline_args = { - "prompt": "CHANGE_ME", - "guidance_scale": 6.0, - "num_inference_steps": 64, - "height": 480, - "width": 848, - "max_sequence_length": 256, - "output_type": "np", -} - -with torch.autocast("cuda", torch.bfloat16) - video = pipe(**pipeline_args).frames[0] +with torch.autocast("cuda", torch.bfloat16): + video = pipe( + prompt="CHANGE_ME", + guidance_scale=6.0, + num_inference_steps=64, + height=480, + width=848, + max_sequence_length=256, + output_type="np" + ).frames[0] export_to_video(video) ```