alkiskoudounas
commited on
Commit
β’
c710dbf
1
Parent(s):
ebb4362
Update README.md
Browse files
README.md
CHANGED
@@ -11,14 +11,14 @@ tags:
|
|
11 |
Model developed for the Unit 1 of the [Diffusion Models Class π§¨](https://github.com/huggingface/diffusion-models-class).
|
12 |
|
13 |
This model is a diffusion model for unconditional image generation of aurora borealis π.
|
14 |
-
It is trained on a small collection aurora pictures and trained for 50 epochs, with π€ Accelerate.
|
15 |
|
16 |
## Usage
|
17 |
|
18 |
```python
|
19 |
from diffusers import DDPMPipeline
|
20 |
|
21 |
-
pipeline = DDPMPipeline.from_pretrained('alkiskoudounas/
|
22 |
image = pipeline().images[0]
|
23 |
image
|
24 |
```
|
|
|
11 |
Model developed for the Unit 1 of the [Diffusion Models Class π§¨](https://github.com/huggingface/diffusion-models-class).
|
12 |
|
13 |
This model is a diffusion model for unconditional image generation of aurora borealis π.
|
14 |
+
It is trained on a small collection of aurora pictures and trained for 50 epochs, with π€ Accelerate.
|
15 |
|
16 |
## Usage
|
17 |
|
18 |
```python
|
19 |
from diffusers import DDPMPipeline
|
20 |
|
21 |
+
pipeline = DDPMPipeline.from_pretrained('alkiskoudounas/sd-aurora-128px')
|
22 |
image = pipeline().images[0]
|
23 |
image
|
24 |
```
|