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