bguisard commited on
Commit
081aeaa
1 Parent(s): a699ab3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -8
README.md CHANGED
@@ -8,11 +8,13 @@ tags:
8
  - diffusers
9
  - jax-diffusers-event
10
  inference: true
 
 
11
  ---
12
 
13
  # Stable Diffusion Nano 2.1
14
 
15
- Stable Diffusion Nano was built during the [JAX/Diffusers community sprint 🧨](jaxsprint).
16
 
17
  Based on stable diffusion and fine-tuned on 128x128 images, Stable Diffusion Nano
18
  allows for fast prototyping of diffusion models, enabling quick experimentation
@@ -21,8 +23,6 @@ with easily available hardware.
21
  It performs reasonably well on several tasks, but it struggles with small details
22
  such as faces.
23
 
24
- jaxsprint: https://github.com/huggingface/community-events/tree/main/jax-controlnet-sprint#jaxdiffusers-community-sprint-
25
-
26
  prompt: A watercolor painting of an otter
27
 
28
  ![images_0)](./images_0.png)
@@ -36,16 +36,14 @@ of focus in background, neon lights
36
 
37
 
38
  ## Training details
39
- All parameters were initialized from the [stabilityai/stable-diffusion-2-1-base](sd21)
40
  model. The unet was fine tuned as follows:
41
 
42
  U-net fine-tuning:
43
  - 200,000 steps, learning rate = 1e-5, batch size = 992 (248 per TPU).
44
  - 100,000 steps, SNR gamma = 5.0, learning rate = 1e-5, batch size = 992 (248 per TPU).
45
- - Trained on [LAION Improved Aesthetics 6plus](laion).
46
 
47
- sd21: https://huggingface.co/stabilityai/stable-diffusion-2-1-base
48
- laion: https://huggingface.co/datasets/ChristophSchuhmann/improved_aesthetics_6plus
49
 
50
  ## License
51
  This model is open access and available to all, with a CreativeML OpenRAIL-M license
@@ -53,4 +51,4 @@ further specifying rights and usage. The CreativeML OpenRAIL License specifies:
53
 
54
  - You can't use the model to deliberately produce nor share illegal or harmful outputs or content.
55
  - The authors claims no rights on the outputs you generate, you are free to use them and are accountable for their use which must not go against the provisions set in the license.
56
- - You may re-distribute the weights and use the model commercially and/or as a service. If you do, please be aware you have to include the same use restrictions as the ones in the license and share a copy of the CreativeML OpenRAIL-M to all your users (please read the license entirely and carefully) Please read the full license here.
 
8
  - diffusers
9
  - jax-diffusers-event
10
  inference: true
11
+ datasets:
12
+ - ChristophSchuhmann/improved_aesthetics_6plus
13
  ---
14
 
15
  # Stable Diffusion Nano 2.1
16
 
17
+ Stable Diffusion Nano was built during the [JAX/Diffusers community sprint 🧨](https://github.com/huggingface/community-events/tree/main/jax-controlnet-sprint#jaxdiffusers-community-sprint-).
18
 
19
  Based on stable diffusion and fine-tuned on 128x128 images, Stable Diffusion Nano
20
  allows for fast prototyping of diffusion models, enabling quick experimentation
 
23
  It performs reasonably well on several tasks, but it struggles with small details
24
  such as faces.
25
 
 
 
26
  prompt: A watercolor painting of an otter
27
 
28
  ![images_0)](./images_0.png)
 
36
 
37
 
38
  ## Training details
39
+ All parameters were initialized from the [stabilityai/stable-diffusion-2-1-base](https://huggingface.co/stabilityai/stable-diffusion-2-1-base)
40
  model. The unet was fine tuned as follows:
41
 
42
  U-net fine-tuning:
43
  - 200,000 steps, learning rate = 1e-5, batch size = 992 (248 per TPU).
44
  - 100,000 steps, SNR gamma = 5.0, learning rate = 1e-5, batch size = 992 (248 per TPU).
45
+ - Trained on [LAION Improved Aesthetics 6plus](https://huggingface.co/datasets/ChristophSchuhmann/improved_aesthetics_6plus).
46
 
 
 
47
 
48
  ## License
49
  This model is open access and available to all, with a CreativeML OpenRAIL-M license
 
51
 
52
  - You can't use the model to deliberately produce nor share illegal or harmful outputs or content.
53
  - The authors claims no rights on the outputs you generate, you are free to use them and are accountable for their use which must not go against the provisions set in the license.
54
+ - You may re-distribute the weights and use the model commercially and/or as a service. If you do, please be aware you have to include the same use restrictions as the ones in the license and share a copy of the CreativeML OpenRAIL-M to all your users (please read the license entirely and carefully) Please read the full license here.