HanLiii commited on
Commit
7329c4e
·
verified ·
1 Parent(s): d159b93

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -4
README.md CHANGED
@@ -3,14 +3,14 @@ license: apache-2.0
3
  ---
4
  Diffusers formation for mochi-1-preview model.
5
 
6
- Use following code to inference
7
 
8
  ```bash
9
  import torch
10
  from diffusers import MochiPipeline
11
  from diffusers.utils import export_to_video
12
 
13
- pipe = MochiPipeline.from_pretrained('HanLiii/mochi-1-preview-diffusers', torch_dtype=torch.float16)
14
  pipe.to("cuda")
15
  prompt = """
16
  A hand with delicate fingers picks up a bright yellow lemon from a wooden bowl
@@ -30,8 +30,10 @@ frames = pipe(prompt,
30
 
31
  export_to_video(frames, "mochi_test.mp4")
32
  ```
33
- Some generated results:
34
- <video controls autoplay src="https://cdn-uploads.huggingface.co/production/uploads/62e34a12c9bece303d146af8/Cm3I6kidy2YP5nu3un7XP.mp4"></video>
 
 
35
 
36
  ---
37
  license: apache-2.0
 
3
  ---
4
  Diffusers formation for mochi-1-preview model.
5
 
6
+ ##Use following code to inference
7
 
8
  ```bash
9
  import torch
10
  from diffusers import MochiPipeline
11
  from diffusers.utils import export_to_video
12
 
13
+ pipe = MochiPipeline.from_pretrained('HanLiii/mochi-1-preview-diffusers', torch_dtype=torch.bfloat16)
14
  pipe.to("cuda")
15
  prompt = """
16
  A hand with delicate fingers picks up a bright yellow lemon from a wooden bowl
 
30
 
31
  export_to_video(frames, "mochi_test.mp4")
32
  ```
33
+ fp16 diffusers version result:
34
+ <video controls autoplay src="https://cdn-uploads.huggingface.co/production/uploads/647d3f9683cdb93baf76f963/0YoDABzUbfErx2McV4LA6.mp4"></video>
35
+ github version result:
36
+ <video controls autoplay src="https://cdn-uploads.huggingface.co/production/uploads/647d3f9683cdb93baf76f963/OY29N7HQK2TVSZcAXQAPx.mp4"></video>
37
 
38
  ---
39
  license: apache-2.0