eolecvk commited on
Commit
c4f09e7
·
1 Parent(s): c15d0cd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -9
README.md CHANGED
@@ -48,13 +48,8 @@ for idx, im in enumerate(images):
48
 
49
  ## Model description
50
 
51
- Trained on 512x512 Avatar character images using 2xA6000 GPUs on [Lambda GPU Cloud](https://lambdalabs.com/service/gpu-cloud) for 700 steps, batch size 4 (about 1 hours, at a cost of about $2).
 
 
 
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