AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Update src/diffusers/pipelines/mochi/pipeline_mochi.py

Browse Source 
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>
This commit is contained in:
Aryan
2024-10-25 14:01:31 +05:30
committed by GitHub
parent 8e11f34a74
commit fb2ede03eb
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/diffusers/pipelines/mochi/pipeline_mochi.py
View File

@@ -48,6 +48,7 @@ EXAMPLE_DOC_STRING = """
```py
>>> import torch
>>> from diffusers import MochiPipeline
>>> from diffusers.utils import export_to_video
>>> pipe = MochiPipeline.from_pretrained("genmo/mochi-1-preview", torch_dtype=torch.bfloat16)
>>> pipe.to("cuda")