Text-to-Image
Diffusers
English
StableDiffusionPipeline
dreambooth
stable-diffusion
stable-diffusion-diffusers
Instructions to use lambda/dreambooth-avatar with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use lambda/dreambooth-avatar with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("lambda/dreambooth-avatar", 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
- Draw Things
- DiffusionBee
Update README.md
Browse files
README.md
CHANGED
|
@@ -48,13 +48,8 @@ for idx, im in enumerate(images):
|
|
| 48 |
|
| 49 |
## Model description
|
| 50 |
|
| 51 |
-
|
|
|
|
|
|
|
|
|
|
| 52 |
|
| 53 |
-
## Links
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
- [Lambda Diffusers](https://github.com/LambdaLabsML/lambda-diffusers)
|
| 57 |
-
- [Model weights in Diffusers format](https://huggingface.co/lambdalabs/sd-naruto-diffusers)
|
| 58 |
-
- [Naruto diffusers repo](https://github.com/eolecvk/naruto-sd)
|
| 59 |
-
|
| 60 |
-
Trained by Eole Cervenka
|
|
|
|
| 48 |
|
| 49 |
## Model description
|
| 50 |
|
| 51 |
+
Base model is Stable Diffusion v1.5 and was trained by Eole Cervenka using Dreambooth with 60 input images sized 512x512 displaying Avatar character images.
|
| 52 |
+
The model is learning to associate Avatar images with the style tokenized as 'avatarart style'.
|
| 53 |
+
Prior preservation was used during training using the class 'Person' to avoid training bleeding into the representations for that class.
|
| 54 |
+
Training ran on 2xA6000 GPUs on [Lambda GPU Cloud](https://lambdalabs.com/service/gpu-cloud) for 700 steps, batch size 4 (a couple hours, at a cost of about $4).
|
| 55 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|