Hugging Face's logo

imnotednamode
/
mochi-1-preview-mix-nf4

Diffusers
Safetensors
4-bit precision
bitsandbytes
Model card Files Community
imnotednamode commited on
Commit
38248f7
•
1 Parent(s): 5289936

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -10,6 +10,7 @@ This mixes mochi with a development version of diffusers to achieve high quality
10
  ```python
11
  from diffusers import MochiPipeline, MochiTransformer3DModel
12
  from diffusers.utils import export_to_video
 
13
  transformer = MochiTransformer3DModel.from_pretrained("imnotednamode/mochi-1-preview-mix-nf4", torch_dtype=torch.bfloat16)
14
  pipe = MochiPipeline.from_pretrained("genmo/mochi-1-preview", revision="refs/pr/18", torch_dtype=torch.bfloat16, transformer=transformer)
15
  pipe.enable_model_cpu_offload()
 
10
  ```python
11
  from diffusers import MochiPipeline, MochiTransformer3DModel
12
  from diffusers.utils import export_to_video
13
+ import torch
14
  transformer = MochiTransformer3DModel.from_pretrained("imnotednamode/mochi-1-preview-mix-nf4", torch_dtype=torch.bfloat16)
15
  pipe = MochiPipeline.from_pretrained("genmo/mochi-1-preview", revision="refs/pr/18", torch_dtype=torch.bfloat16, transformer=transformer)
16
  pipe.enable_model_cpu_offload()