Instructions to use yresearch/swd-medium-6-steps with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use yresearch/swd-medium-6-steps with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("yresearch/swd-medium-6-steps", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
Update README.md
Browse files
README.md
CHANGED
|
@@ -29,8 +29,8 @@ pipe.transformer = PeftModel.from_pretrained(
|
|
| 29 |
lora_path,
|
| 30 |
)
|
| 31 |
generator = torch.Generator().manual_seed(0)
|
| 32 |
-
prompt = '
|
| 33 |
-
sigmas = [1.0000, 0.9454, 0.8959, 0.7904, 0.7371, 0.6022]
|
| 34 |
scales = [32, 48, 64, 80, 96, 128]
|
| 35 |
|
| 36 |
images = run(
|
|
@@ -47,7 +47,7 @@ images = run(
|
|
| 47 |
).images
|
| 48 |
```
|
| 49 |
<p align="center">
|
| 50 |
-
<img src="
|
| 51 |
</p>
|
| 52 |
|
| 53 |
## 🔧 Training
|
|
|
|
| 29 |
lora_path,
|
| 30 |
)
|
| 31 |
generator = torch.Generator().manual_seed(0)
|
| 32 |
+
prompt = 'A gold astronaut meditating in a lush green forest by a lake'
|
| 33 |
+
sigmas = [1.0000, 0.9454, 0.8959, 0.7904, 0.7371, 0.6022, 0.0000]
|
| 34 |
scales = [32, 48, 64, 80, 96, 128]
|
| 35 |
|
| 36 |
images = run(
|
|
|
|
| 47 |
).images
|
| 48 |
```
|
| 49 |
<p align="center">
|
| 50 |
+
<img src="img.jpg" width="512px"/>
|
| 51 |
</p>
|
| 52 |
|
| 53 |
## 🔧 Training
|