Upload bocchi-the-rock-character with huggingface_hub
Browse files- bocchi-the-rock-character/README.md +96 -0
- bocchi-the-rock-character/bocchitherock24000steps.ckpt +3 -0
- bocchi-the-rock-character/bocchitherock24000stepspruned.ckpt +3 -0
- bocchi-the-rock-character/bocchitherock48000steps.ckpt +3 -0
- bocchi-the-rock-character/bocchitherock48000stepspruned.ckpt +3 -0
- bocchi-the-rock-character/bocchitherock50000stepspruned.ckpt +3 -0
- bocchi-the-rock-character/examples/failure/4265343039-1470057774.png +0 -0
- bocchi-the-rock-character/examples/failure/4265343048-2869643584.png +0 -0
- bocchi-the-rock-character/examples/failure/4265343089-2940163958.png +0 -0
- bocchi-the-rock-character/examples/failure/4265343091-129639375.png +0 -0
- bocchi-the-rock-character/examples/grids/xy_grid-0028-24.jpg +0 -0
- bocchi-the-rock-character/examples/grids/xy_grid-0029-24.jpg +0 -0
- bocchi-the-rock-character/examples/grids/xy_grid-0030-24.jpg +0 -0
- bocchi-the-rock-character/examples/with_inpaint/4265343022-3534836762.png +0 -0
- bocchi-the-rock-character/examples/with_inpaint/4265343062-1047638199.png +0 -0
- bocchi-the-rock-character/examples/with_inpaint/4265343066-3979275255.png +0 -0
- bocchi-the-rock-character/examples/with_inpaint/4265343068-2420755431.png +0 -0
- bocchi-the-rock-character/examples/without_inpaint/4265343016-2842516738.png +0 -0
- bocchi-the-rock-character/examples/without_inpaint/4265343053-918713189.png +0 -0
- bocchi-the-rock-character/examples/without_inpaint/4265343054-2839948768.png +0 -0
- bocchi-the-rock-character/examples/without_inpaint/4265343083-1372779456.png +0 -0
- bocchi-the-rock-character/examples/without_inpaint/4265343084-3548261345.png +0 -0
- bocchi-the-rock-character/examples/without_inpaint/4265343086-2648280139.png +0 -0
- bocchi-the-rock-character/examples/without_inpaint/4265343092-803155289.png +0 -0
- bocchi-the-rock-character/examples/without_inpaint/4265343096-399054050.png +0 -0
- 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 |
+

|
| 17 |
+
|
| 18 |
+
Without inpainting
|
| 19 |
+

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

|
| 28 |
+

|
| 29 |
+

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

|
| 78 |
+

|
| 79 |
+

|
| 80 |
+
|
| 81 |
+
|
| 82 |
+
Without inpainting
|
| 83 |
+

|
| 84 |
+

|
| 85 |
+

|
| 86 |
+

|
| 87 |
+

|
| 88 |
+

|
| 89 |
+

|
| 90 |
+

|
| 91 |
+
|
| 92 |
+
Some failure cases
|
| 93 |
+

|
| 94 |
+

|
| 95 |
+

|
| 96 |
+

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