carlosabadia
commited on
Commit
•
cbfa886
1
Parent(s):
ce17b19
Update README.md
Browse files
README.md
CHANGED
@@ -36,9 +36,18 @@ Model supported in a [Gradio](https://github.com/gradio-app/gradio) Web UI and C
|
|
36 |
## Usage
|
37 |
|
38 |
```python
|
39 |
-
from diffusers import StableDiffusionPipeline
|
|
|
40 |
|
41 |
-
|
42 |
-
|
43 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
44 |
```
|
|
|
36 |
## Usage
|
37 |
|
38 |
```python
|
39 |
+
from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
|
40 |
+
import torch
|
41 |
|
42 |
+
model_id = "carlosabadia/hasbulla"
|
43 |
+
|
44 |
+
# Use the DPMSolverMultistepScheduler (DPM-Solver++) scheduler here instead
|
45 |
+
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
46 |
+
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
47 |
+
pipe = pipe.to("cuda")
|
48 |
+
|
49 |
+
prompt = "a portrait of hasbulla person"
|
50 |
+
image = pipe(prompt).images[0]
|
51 |
+
|
52 |
+
image.save("hasbulla.png")
|
53 |
```
|