Luo-Yihong
commited on
Commit
•
5efa73d
1
Parent(s):
f032160
Update README.md
Browse files
README.md
CHANGED
@@ -77,6 +77,27 @@ imgs
|
|
77 |
```
|
78 |
![man](man.jpg)
|
79 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
80 |
You may try some interesting applications, like:
|
81 |
```python
|
82 |
generator = torch.manual_seed(318)
|
|
|
77 |
```
|
78 |
![man](man.jpg)
|
79 |
|
80 |
+
Moreover, it is observed that when combined with new base models, our YOSO-LoRA is able to use some advanced ode-solvers:
|
81 |
+
```python
|
82 |
+
import torch
|
83 |
+
from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
|
84 |
+
pipeline = DiffusionPipeline.from_pretrained("stablediffusionapi/realistic-vision-v51", torch_dtype = torch.float16)
|
85 |
+
pipeline = pipeline.to('cuda')
|
86 |
+
pipeline.scheduler = DPMSolverMultistepScheduler.from_pretrained("runwayml/stable-diffusion-v1-5", subfolder="scheduler")
|
87 |
+
generator = torch.manual_seed(323)
|
88 |
+
steps = 2
|
89 |
+
imgs= pipeline(prompt="A photo of a girl, XT3",
|
90 |
+
num_inference_steps=steps,
|
91 |
+
num_images_per_prompt = 1,
|
92 |
+
generator = generator,
|
93 |
+
guidance_scale=1.5,
|
94 |
+
)[0]
|
95 |
+
imgs[0]
|
96 |
+
```
|
97 |
+
![girl](girl.jpg)
|
98 |
+
|
99 |
+
We encourage you to experiment with various solvers to obtain better samples. We will try to improve the compatibility of the YOSO-LoRA with different solvers.
|
100 |
+
|
101 |
You may try some interesting applications, like:
|
102 |
```python
|
103 |
generator = torch.manual_seed(318)
|