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

Merge pull request #104 from a-r-r-o-w/fix-format-specifier

Browse Source 
fix format specifier.
This commit is contained in:
Sayak Paul
2024-11-29 16:29:58 +05:30
committed by GitHub
parent d5cc7c66f1 c8a118cefd
commit 099c9c35c4
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.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)
```