alkiskoudounas commited on
Commit
c710dbf
β€’
1 Parent(s): ebb4362

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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/stable-diffusion-aurora-128px')
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
  ```