imnotednamode
commited on
Commit
•
91a72e1
1
Parent(s):
e0ae284
Update README.md
Browse files
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()
|