ttttdiva commited on
Commit
52b2292
·
1 Parent(s): 0bf04d9

Upload bocchi-the-rock-character with huggingface_hub

Browse files
Files changed (26) hide show
  1. bocchi-the-rock-character/README.md +96 -0
  2. bocchi-the-rock-character/bocchitherock24000steps.ckpt +3 -0
  3. bocchi-the-rock-character/bocchitherock24000stepspruned.ckpt +3 -0
  4. bocchi-the-rock-character/bocchitherock48000steps.ckpt +3 -0
  5. bocchi-the-rock-character/bocchitherock48000stepspruned.ckpt +3 -0
  6. bocchi-the-rock-character/bocchitherock50000stepspruned.ckpt +3 -0
  7. bocchi-the-rock-character/examples/failure/4265343039-1470057774.png +0 -0
  8. bocchi-the-rock-character/examples/failure/4265343048-2869643584.png +0 -0
  9. bocchi-the-rock-character/examples/failure/4265343089-2940163958.png +0 -0
  10. bocchi-the-rock-character/examples/failure/4265343091-129639375.png +0 -0
  11. bocchi-the-rock-character/examples/grids/xy_grid-0028-24.jpg +0 -0
  12. bocchi-the-rock-character/examples/grids/xy_grid-0029-24.jpg +0 -0
  13. bocchi-the-rock-character/examples/grids/xy_grid-0030-24.jpg +0 -0
  14. bocchi-the-rock-character/examples/with_inpaint/4265343022-3534836762.png +0 -0
  15. bocchi-the-rock-character/examples/with_inpaint/4265343062-1047638199.png +0 -0
  16. bocchi-the-rock-character/examples/with_inpaint/4265343066-3979275255.png +0 -0
  17. bocchi-the-rock-character/examples/with_inpaint/4265343068-2420755431.png +0 -0
  18. bocchi-the-rock-character/examples/without_inpaint/4265343016-2842516738.png +0 -0
  19. bocchi-the-rock-character/examples/without_inpaint/4265343053-918713189.png +0 -0
  20. bocchi-the-rock-character/examples/without_inpaint/4265343054-2839948768.png +0 -0
  21. bocchi-the-rock-character/examples/without_inpaint/4265343083-1372779456.png +0 -0
  22. bocchi-the-rock-character/examples/without_inpaint/4265343084-3548261345.png +0 -0
  23. bocchi-the-rock-character/examples/without_inpaint/4265343086-2648280139.png +0 -0
  24. bocchi-the-rock-character/examples/without_inpaint/4265343092-803155289.png +0 -0
  25. bocchi-the-rock-character/examples/without_inpaint/4265343096-399054050.png +0 -0
  26. bocchi-the-rock-character/examples/without_inpaint/4265343100-3858388158.png +0 -0
bocchi-the-rock-character/README.md ADDED
@@ -0,0 +1,96 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: creativeml-openrail-m
3
+ ---
4
+
5
+ ---
6
+ license: creativeml-openrail-m
7
+ ---
8
+
9
+ This is a low-quality bocchi-the-rock (ぼっち・ざ・ろっく!) character model.
10
+ Similar to my [yama-no-susume model](https://huggingface.co/alea31415/yama-no-susume), this model is capable of generating **multi-character scenes** beyond images of a single character.
11
+ Of course, the result is still hit-or-miss, but I with some chance you can get the entire Kessoku Band right in one shot,
12
+ and otherwise, you can always rely on inpainting.
13
+ Here are two examples:
14
+
15
+ With inpainting
16
+ ![4265343062-1047638199](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/with_inpaint/4265343062-1047638199.png)
17
+
18
+ Without inpainting
19
+ ![4265343086-2648280139](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/without_inpaint/4265343086-2648280139.png)
20
+
21
+
22
+ ### Characters
23
+
24
+ The model knows 12 characters from bocchi the rock.
25
+ The ressemblance with a character can be improved by a better description of their appearance (for example by adding long wavy hair to ShimizuEliza).
26
+
27
+ ![xy_grid-0028-24](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/grids/xy_grid-0028-24.jpg)
28
+ ![xy_grid-0029-24](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/grids/xy_grid-0029-24.jpg)
29
+ ![xy_grid-0030-24](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/grids/xy_grid-0030-24.jpg)
30
+
31
+
32
+ ### Dataset description
33
+
34
+ The dataset contains around 27K images with the following composition
35
+ - 7024 anime screenshots
36
+ - 1630 fan arts
37
+ - 18519 customized regularization images
38
+
39
+ The model is trained with a specific weighting scheme to balance between different concepts.
40
+ For example, the above three categories have weights respectively 0.3, 0.25, and 0.45.
41
+ Each category is itself split into many sub-categories in a hierarchical way.
42
+ For more details on the data preparation process please refer to https://github.com/cyber-meow/anime_screenshot_pipeline
43
+
44
+
45
+ ### Training Details
46
+
47
+ #### Trainer
48
+ The model is trained using [EveryDream1](https://github.com/victorchall/EveryDream-trainer) as
49
+ EveryDream seems to be the only trainer out there that supports sample weighting (through the use of `multiply.txt`).
50
+ Note that for future training it makes sense to migrate to [EveryDream2](https://github.com/victorchall/EveryDream2trainer).
51
+
52
+ #### Hardware and cost
53
+ The model is trained on runpod using 3090 and cost me around 15 dollors.
54
+
55
+ #### Hyperparameter specification
56
+
57
+ The model is trained for 50000 steps, at batch size 4, lr 1e-6, resolution 512, and conditional dropping rate of 10%.
58
+
59
+ Note that as a consequence of the weighting scheme which translates into a number of different multiply for each image,
60
+ the count of repeat and epoch has a quite different meaning here.
61
+ For example, depending on the weighting, I have around 300K images (some images are used multiple times) in an epoch,
62
+ and therefore I did not even finish an entire epoch with the 50000 steps at batch size 4.
63
+
64
+ ### Failures
65
+
66
+ - For the first 24000 steps I use the trigger words `Bfan1` and `Bfan2` for the two fans of Bocchi.
67
+ However, these two words are too similar and the model fails to different characters for these.
68
+ Therefore I changed Bfan2 to Bofa2 at step 24000. This seemed to solve the problem.
69
+ - Character blending is always an issue.
70
+ - When prompting the four characters of Kessoku Band we often get side shots.
71
+ I think this is because of some overfitting to a particular image.
72
+
73
+
74
+ ### More Example Generations
75
+
76
+ With inpainting
77
+ ![4265343068-2420755431](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/with_inpaint/4265343068-2420755431.png)
78
+ ![4265343066-3979275255](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/with_inpaint/4265343066-3979275255.png)
79
+ ![4265343022-3534836762](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/with_inpaint/4265343022-3534836762.png)
80
+
81
+
82
+ Without inpainting
83
+ ![4265343092-803155289](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/without_inpaint/4265343092-803155289.png)
84
+ ![4265343053-918713189](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/without_inpaint/4265343053-918713189.png)
85
+ ![4265343054-2839948768](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/without_inpaint/4265343054-2839948768.png)
86
+ ![4265343096-399054050](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/without_inpaint/4265343096-399054050.png)
87
+ ![4265343100-3858388158](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/without_inpaint/4265343100-3858388158.png)
88
+ ![4265343016-2842516738](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/without_inpaint/4265343016-2842516738.png)
89
+ ![4265343084-3548261345](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/without_inpaint/4265343084-3548261345.png)
90
+ ![4265343083-1372779456](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/without_inpaint/4265343083-1372779456.png)
91
+
92
+ Some failure cases
93
+ ![4265343089-2940163958](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/failure/4265343089-2940163958.png)
94
+ ![4265343091-129639375](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/failure/4265343091-129639375.png)
95
+ ![4265343048-2869643584](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/failure/4265343048-2869643584.png)
96
+ ![4265343039-1470057774](https://huggingface.co/alea31415/bocchi-the-rock-character/resolve/main/examples/failure/4265343039-1470057774.png)
bocchi-the-rock-character/bocchitherock24000steps.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:736c25fa26cf3c5f124878a971758fc1b3691696509deaa4ab4e31aa7a75b30a
3
+ size 12126929195
bocchi-the-rock-character/bocchitherock24000stepspruned.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:85bb016f52bad19fe69a40b2f5b6e795fd46a076abe0d5bd189b3ffb5c19dc0e
3
+ size 2132892617
bocchi-the-rock-character/bocchitherock48000steps.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:97279026843d9b7bcaa093a8818f1d4ebf227115d92efebe1d35c3729f350523
3
+ size 12126929195
bocchi-the-rock-character/bocchitherock48000stepspruned.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7b7755321cb0357e2575fdaf2efb20e5640dfa0e8dfe294541842e5cc24fd56c
3
+ size 2132892617
bocchi-the-rock-character/bocchitherock50000stepspruned.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3fc93a9be90bd5d7ef7fb385f20e955fd5dabc15d2fb0064dbb1c6db8f79166
3
+ size 2132892617
bocchi-the-rock-character/examples/failure/4265343039-1470057774.png ADDED
bocchi-the-rock-character/examples/failure/4265343048-2869643584.png ADDED
bocchi-the-rock-character/examples/failure/4265343089-2940163958.png ADDED
bocchi-the-rock-character/examples/failure/4265343091-129639375.png ADDED
bocchi-the-rock-character/examples/grids/xy_grid-0028-24.jpg ADDED
bocchi-the-rock-character/examples/grids/xy_grid-0029-24.jpg ADDED
bocchi-the-rock-character/examples/grids/xy_grid-0030-24.jpg ADDED
bocchi-the-rock-character/examples/with_inpaint/4265343022-3534836762.png ADDED
bocchi-the-rock-character/examples/with_inpaint/4265343062-1047638199.png ADDED
bocchi-the-rock-character/examples/with_inpaint/4265343066-3979275255.png ADDED
bocchi-the-rock-character/examples/with_inpaint/4265343068-2420755431.png ADDED
bocchi-the-rock-character/examples/without_inpaint/4265343016-2842516738.png ADDED
bocchi-the-rock-character/examples/without_inpaint/4265343053-918713189.png ADDED
bocchi-the-rock-character/examples/without_inpaint/4265343054-2839948768.png ADDED
bocchi-the-rock-character/examples/without_inpaint/4265343083-1372779456.png ADDED
bocchi-the-rock-character/examples/without_inpaint/4265343084-3548261345.png ADDED
bocchi-the-rock-character/examples/without_inpaint/4265343086-2648280139.png ADDED
bocchi-the-rock-character/examples/without_inpaint/4265343092-803155289.png ADDED
bocchi-the-rock-character/examples/without_inpaint/4265343096-399054050.png ADDED
bocchi-the-rock-character/examples/without_inpaint/4265343100-3858388158.png ADDED