feizhengcong's picture
Update README.md
52d9fc4 verified
|
raw
history blame
742 Bytes
metadata
license: apache-2.0

Diffusers formation for mochi-1-preview model.

It was create by scripts: https://github.com/huggingface/diffusers/blob/mochi/scripts/convert_mochi_to_diffusers.py

The model can be directly load from pretrained with mochi branch: https://github.com/huggingface/diffusers/tree/mochi

from diffusers import MochiPipeline
from diffusers.utils import export_to_video

pipe = MochiPipeline.from_pretrained(path, torch_dtype=torch.bfloat16)
pipe.to("cuda")
prompt = "Close-up of a chameleon's eye, with its scaly skin changing color. Ultra high resolution 4k."
frames = pipe(prompt, num_inference_steps=28, guidance_scale=3.5).frames[0]

export_to_video(frames, "mochi.mp4")

license: apache-2.0