BAAI
/

Simplify diffusers example

#3
Files changed (1) hide show
  1. README.md +8 -2
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("../AltDiffusion", torch_dtype=torch.float16)
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)