Hugging Face's logo

imnotednamode
/
mochi-1-preview-mix-nf4-small

Diffusers
Safetensors
4-bit precision
bitsandbytes
Model card Files Community
1
imnotednamode commited on
Commit
4f96653
•
1 Parent(s): 7b31048

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -29,6 +29,7 @@ To use:
29
  ```
30
  from diffusers import MochiPipeline, MochiTransformer3DModel
31
  from diffusers.utils import export_to_video
 
32
  transformer = MochiTransformer3DModel.from_pretrained("imnotednamode/mochi-1-preview-mix-nf4-small", torch_dtype=torch.float16)
33
  pipe = MochiPipeline.from_pretrained("genmo/mochi-1-preview", revision="refs/pr/18", torch_dtype=torch.float16, transformer=transformer)
34
  pipe.enable_model_cpu_offload()
 
29
  ```
30
  from diffusers import MochiPipeline, MochiTransformer3DModel
31
  from diffusers.utils import export_to_video
32
+ import torch
33
  transformer = MochiTransformer3DModel.from_pretrained("imnotednamode/mochi-1-preview-mix-nf4-small", torch_dtype=torch.float16)
34
  pipe = MochiPipeline.from_pretrained("genmo/mochi-1-preview", revision="refs/pr/18", torch_dtype=torch.float16, transformer=transformer)
35
  pipe.enable_model_cpu_offload()