Simplify diffusers example
#3
by
patrickvonplaten
- opened
README.md
CHANGED
|
@@ -72,12 +72,18 @@ You can see the documentation page [here](https://huggingface.co/docs/diffusers/
|
|
| 72 |
|
| 73 |
The following example will use the fast DPM scheduler to generate an image in ca. 2 seconds on a V100.
|
| 74 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 75 |
```python
|
| 76 |
-
!pip install git+https://github.com/huggingface/diffusers.git torch transformers
|
| 77 |
from diffusers import AltDiffusionPipeline, DPMSolverMultistepScheduler
|
| 78 |
import torch
|
| 79 |
|
| 80 |
-
pipe = AltDiffusionPipeline.from_pretrained("
|
| 81 |
pipe = pipe.to("cuda")
|
| 82 |
|
| 83 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
|
|
|
| 72 |
|
| 73 |
The following example will use the fast DPM scheduler to generate an image in ca. 2 seconds on a V100.
|
| 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:
|
| 81 |
+
|
| 82 |
```python
|
|
|
|
| 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)
|