imnotednamode commited on
Commit
91a72e1
1 Parent(s): e0ae284

Update README.md

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