multimodalart HF staff commited on
Commit
3439518
1 Parent(s): ef7c178

Faster diffusers demo

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -74,7 +74,7 @@ The following example will use the fast DPM scheduler to generate an image in ca
74
 
75
  First you should install diffusers main branch and some dependencies:
76
  ```
77
- pip install git+https://github.com/huggingface/diffusers.git torch transformers sentencepiece
78
  ```
79
 
80
  then you can run the following example:
@@ -83,7 +83,7 @@ then you can run the following example:
83
  from diffusers import AltDiffusionPipeline, DPMSolverMultistepScheduler
84
  import torch
85
 
86
- pipe = AltDiffusionPipeline.from_pretrained("BAAI/AltDiffusion", torch_dtype=torch.float16)
87
  pipe = pipe.to("cuda")
88
 
89
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 
74
 
75
  First you should install diffusers main branch and some dependencies:
76
  ```
77
+ pip install git+https://github.com/huggingface/diffusers.git torch transformers accelerate sentencepiece
78
  ```
79
 
80
  then you can run the following example:
 
83
  from diffusers import AltDiffusionPipeline, DPMSolverMultistepScheduler
84
  import torch
85
 
86
+ pipe = AltDiffusionPipeline.from_pretrained("BAAI/AltDiffusion", torch_dtype=torch.float16, revision="fp16")
87
  pipe = pipe.to("cuda")
88
 
89
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)