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 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|