|
import torch |
|
from diffusers import MochiPipeline |
|
from diffusers.utils import export_to_video |
|
import gradio as gr |
|
pipe = MochiPipeline.from_pretrained("genmo/mochi-1-preview", variant="bf16", torch_dtype=torch.bfloat16) |
|
|
|
|
|
pipe.enable_model_cpu_offload() |
|
pipe.enable_vae_tiling() |
|
|
|
prompt = "Close-up of a chameleon's eye, with its scaly skin changing color. Ultra high resolution 4k." |
|
frames = pipe(prompt, num_frames=84).frames[0] |
|
|
|
export_to_video(frames, "mochi.mp4", fps=30) |
|
|
|
with gr.Blocks() as b: |
|
vid=gr.Video("mochi.mp4") |
|
b.launch() |
