storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-20 07:45:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix format specifier.

Browse Source
This commit is contained in:
sayakpaul
2024-11-29 16:23:14 +05:30
parent d5cc7c66f1
commit c8a118cefd
1 changed files with 10 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training/mochi-1/text_to_video_lora.py
+10 -12
View File
@@ -110,18 +110,16 @@ pipe = MochiPipeline.from_pretrained("genmo/mochi-1-preview")
pipe.load_lora_weights("CHANGE_ME") pipe.load_lora_weights("CHANGE_ME")
pipe.enable_model_cpu_offload() pipe.enable_model_cpu_offload()
pipeline_args = { with torch.autocast("cuda", torch.bfloat16):
"prompt": "CHANGE_ME", video = pipe(
"guidance_scale": 6.0, prompt="CHANGE_ME",
"num_inference_steps": 64, guidance_scale=6.0,
"height": 480, num_inference_steps=64,
"width": 848, height=480,
"max_sequence_length": 256, width=848,
"output_type": "np", max_sequence_length=256,
} output_type="np"
).frames[0]
with torch.autocast("cuda", torch.bfloat16)
video = pipe(**pipeline_args).frames[0]
export_to_video(video) export_to_video(video)
``` ```