imnotednamode
/

mochi-1-preview-mix-nf4

4-bit precision

Model card Files Files and versions Community

imnotednamode commited on 3 days ago

Commit

5289936

•

1 Parent(s): 96ed778

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -11,7 +11,7 @@ This mixes mochi with a development version of diffusers to achieve high quality
 from diffusers import MochiPipeline, MochiTransformer3DModel
 from diffusers.utils import export_to_video
 transformer = MochiTransformer3DModel.from_pretrained("imnotednamode/mochi-1-preview-mix-nf4", torch_dtype=torch.bfloat16)
-pipe = MochiPipeline.from_pretrained("genmo/mochi-1-preview", variant="refs/pr/18", torch_dtype=torch.bfloat16, transformer=transformer)
 pipe.enable_model_cpu_offload()
 pipe.enable_vae_tiling()
 frames = pipe("A camera follows a squirrel running around on a tree branch", num_inference_steps=100, guidance_scale=4.5, height=480, width=848, num_frames=161).frames[0]

 from diffusers import MochiPipeline, MochiTransformer3DModel
 from diffusers.utils import export_to_video
 transformer = MochiTransformer3DModel.from_pretrained("imnotednamode/mochi-1-preview-mix-nf4", torch_dtype=torch.bfloat16)
+pipe = MochiPipeline.from_pretrained("genmo/mochi-1-preview", revision="refs/pr/18", torch_dtype=torch.bfloat16, transformer=transformer)
 pipe.enable_model_cpu_offload()
 pipe.enable_vae_tiling()
 frames = pipe("A camera follows a squirrel running around on a tree branch", num_inference_steps=100, guidance_scale=4.5, height=480, width=848, num_frames=161).frames[0]