File size: 4,664 Bytes
514db2a
 
d926073
 
 
 
 
 
 
 
 
 
cba2c16
 
 
 
 
 
d926073
4581d83
 
 
 
 
 
 
00e069b
 
b7544f4
609d705
377440c
609d705
b060779
 
 
 
2e25e36
7fd33cb
 
 
609d705
 
2e25e36
 
 
 
609d705
377440c
609d705
cca89ef
5bc6274
cca89ef
609d705
dcb0179
609d705
dcb0179
 
 
b7544f4
5bc6274
609d705
 
377440c
609d705
4581d83
 
 
f091ad0
4581d83
 
 
8ca8c6c
 
 
4327b81
4581d83
 
6bcf28d
4327b81
a6da21e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
---
license: mit
datasets:
- hearmeneigh/e621-rising-v3-curated
- hearmeneigh/e621-rising-v3-finetuner
library_name: diffusers
pipeline_tag: text-to-image
tags:
- anthro
- furry
- e621
- nsfw
- booru
- imagebooru
- imageboard
- gelbooru
- danbooru
- rule34
- not-for-all-audiences
---

<div style='background: #ffeef1; border: 1px solid #fd91a4; padding:1em; border-radius:3px; margin-bottom:2em;'>
<h3 style='margin:0'>NSFW</h3>
<p style='margin:0'>This model is not suitable for use by minors. The model can and will produce X-rated/NFSW content.</p>
</div>


<div style='background: #eefaff; border: 1px solid #91cefd; padding:1em; border-radius:3px; margin-bottom:2em;'>
<h1 style='margin:0'>Quickstart</h1>
<div style='margin:0; margin-top: 1em; margin-left:1em;'>
  <h3 style='margin:0'>Downloads</h3>
    <div style="margin-left: 1em;">
      <a href="https://huggingface.co/hearmeneigh/e621-rising-v3-safetensors/resolve/main/e621-rising-v3-epoch-27.fp16.safetensors" style="text-decoration: none !important;">

      </a>
      <a href="https://huggingface.co/hearmeneigh/e621-rising-v3-safetensors/resolve/main/e621-rising-v3-epoch-27.fp16.safetensors" style="text-decoration: underline">Checkpoint</a>
      <small style='padding-left: 0.5em'>
        (<a href="https://huggingface.co/hearmeneigh/e621-rising-v3-safetensors/resolve/main/e621-rising-v3-epoch-27.safetensors">fp32</a> | 
        <a href="https://huggingface.co/hearmeneigh/e621-rising-v3-safetensors/resolve/main/e621-rising-v3-epoch-27.fp16.safetensors">fp16</a> | 
        <a href="https://huggingface.co/hearmeneigh/e621-rising-v3-safetensors/resolve/main/e621-rising-v3-epoch-27.bf16.safetensors">bf16</a>)
      </small>
      <br />
      <a href="https://huggingface.co/datasets/hearmeneigh/e621-rising-v3-preliminary-data/resolve/main/rising-v3.csv" style="text-decoration: none !important">

      </a>
      <a href="https://huggingface.co/datasets/hearmeneigh/e621-rising-v3-preliminary-data/resolve/main/rising-v3.csv">Tag Autocomplete CSV</a>
    </div>
  <h3 style='margin:0; margin-top: 1em;'>Reference</h3>
    <div style='margin-left: 1em;'>
      • <a href="https://huggingface.co/hearmeneigh/e621-rising-v3/blob/main/INSTALL.md">Installation instructions</a>
      <br />
      • <a href="https://huggingface.co/hearmeneigh/e621-rising-v3/blob/main/CHANGELOG.md">What's new in v3?</a>
      <br />
      • <a href="https://huggingface.co/hearmeneigh/e621-rising-v3/blob/main/PROMPT-EXAMPLES.md" style='display: inline-block; margin-top:0.5em'>Prompt examples</a>
      <br />
      • <a href="https://huggingface.co/hearmeneigh/e621-rising-v3/blob/main/PROMPT-GUIDE.md">Prompt guide</a>
      <br />
      • <a href="https://huggingface.co/datasets/hearmeneigh/e621-rising-v3-preliminary-data/blob/main/tag-counts.by-name.json" style='display: inline-block; margin-top:0.5em'>Tag list</a>
      <br />
      • <a href="https://huggingface.co/hearmeneigh/e621-rising-v3/blob/main/AUTOCOMPLETE.md">Tag autocomplete guide</a>
    </div>
</div>
</div>

# E621 Rising V3 (SDXL)
* Furry / anthro base model trained with images (mainly) from [E621](https://e621.net) 
* Guaranteed **NSFW** or your money back
* Stable Diffusion XL 1.0 model:
  * `1024x1024px`
  * Trained with 11 epochs of 280,000 images each
  * Finetuned with 16 epochs of 40,000 images each
* Compatible with [🤗 `diffusers`](https://huggingface.co/docs/diffusers/index)
* Compatible with [`stable-diffusion-webui`](https://github.com/AUTOMATIC1111/stable-diffusion-webui)
* Fully open source crawl, dataset, curation, and training process:
  * Use these tools to train your own version with your own dataset!
  * [Configuration](https://github.com/hearmeneigh/e621-rising-configs)
  * [Toolchain](https://github.com/hearmeneigh/dataset-rising)
  * [Dataset](https://huggingface.co/datasets/hearmeneigh/e621-rising-v3-curated)


## Training Procedure

[Training legend](TBD)

* 160 images per batch (epoch variant)
* `1024x1024px` image size
* Adam optimizer
  * Beta1 = `0.9`
  * Beta2 = `0.999`
  * Weight decay = `1e-2`
  * Epsilon = `1e-08`
* Constant learning rate `4e-6`
* `fp16` mixed precision
* SNR gamma set to `5.0`
* Noise offset set to `0.07`
* `cosine_with_restarts` scheduler
* 11 epochs of [V3 curated dataset](https://huggingface.co/datasets/hearmeneigh/e621-rising-v3-curated) samples resized to `< 1024x1024px` (maintain aspect ratio)
* 16 epochs of [V3 finetuner dataset](https://huggingface.co/datasets/hearmeneigh/e621-rising-v3-finetuner) samples resized to `< 1024x1024px` (maintain aspect ratio)
* Tags for each sample are shuffled for each epoch