Text-to-Image
Diffusers
Safetensors
StableDiffusionPipeline
stable-diffusion
stable-diffusion-diffusers
jax-diffusers-event
Instructions to use bguisard/stable-diffusion-nano-2-1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use bguisard/stable-diffusion-nano-2-1 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("bguisard/stable-diffusion-nano-2-1", torch_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
Bruno Guisard commited on
Commit ·
a699ab3
1
Parent(s): 096f80b
Update README
Browse files
README.md
CHANGED
|
@@ -12,17 +12,32 @@ inference: true
|
|
| 12 |
|
| 13 |
# Stable Diffusion Nano 2.1
|
| 14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
prompt: A watercolor painting of an otter
|
| 16 |
|
| 17 |

|
| 18 |
|
| 19 |
-
prompt: Marvel MCU deadpool, red mask, red shirt, red gloves, black shoulders,
|
|
|
|
|
|
|
|
|
|
| 20 |
|
| 21 |

|
| 22 |
|
| 23 |
|
| 24 |
## Training details
|
| 25 |
-
All parameters were initialized from the [stabilityai/stable-diffusion-2-1-base](sd21)
|
|
|
|
| 26 |
|
| 27 |
U-net fine-tuning:
|
| 28 |
- 200,000 steps, learning rate = 1e-5, batch size = 992 (248 per TPU).
|
|
@@ -33,7 +48,8 @@ sd21: https://huggingface.co/stabilityai/stable-diffusion-2-1-base
|
|
| 33 |
laion: https://huggingface.co/datasets/ChristophSchuhmann/improved_aesthetics_6plus
|
| 34 |
|
| 35 |
## License
|
| 36 |
-
This model is open access and available to all, with a CreativeML OpenRAIL-M license
|
|
|
|
| 37 |
|
| 38 |
- You can't use the model to deliberately produce nor share illegal or harmful outputs or content.
|
| 39 |
- 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.
|
|
|
|
| 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
|
| 19 |
+
with easily available hardware.
|
| 20 |
+
|
| 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 |

|
| 29 |
|
| 30 |
+
prompt: Marvel MCU deadpool, red mask, red shirt, red gloves, black shoulders,
|
| 31 |
+
black elbow pads, black legs, gold buckle, black belt, black mask, white eyes,
|
| 32 |
+
black boots, fuji low light color 35mm film, downtown Osaka alley at night out
|
| 33 |
+
of focus in background, neon lights
|
| 34 |
|
| 35 |

|
| 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).
|
|
|
|
| 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
|
| 52 |
+
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.
|