2601
Browse files- media/result_grid2.jpg +2 -2
- samples/sdxs_sdxl2_192x320_0.jpg +2 -2
- samples/sdxs_sdxl2_256x320_0.jpg +2 -2
- samples/sdxs_sdxl2_320x192_0.jpg +2 -2
- samples/sdxs_sdxl2_320x256_0.jpg +2 -2
- samples/sdxs_sdxl2_320x320_0.jpg +2 -2
- test.ipynb +2 -2
- train.py +4 -4
- unet/config.json +2 -2
- unet/diffusion_pytorch_model.safetensors +2 -2
- {sdxs_sdxl2 → unet_old}/config.json +2 -2
- {sdxs_sdxl2 → unet_old}/diffusion_pytorch_model.safetensors +2 -2
media/result_grid2.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/sdxs_sdxl2_192x320_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/sdxs_sdxl2_256x320_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/sdxs_sdxl2_320x192_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/sdxs_sdxl2_320x256_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/sdxs_sdxl2_320x320_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
test.ipynb
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:20f9413ba8b38673409699a49f00a6f9794a0c5153625b422b6151af0df9a940
|
| 3 |
+
size 2047294
|
train.py
CHANGED
|
@@ -28,10 +28,10 @@ from transformers import AutoTokenizer, AutoModel
|
|
| 28 |
|
| 29 |
# --------------------------- Параметры ---------------------------
|
| 30 |
ds_path = "/workspace/sdxs/datasets/mjnj"
|
| 31 |
-
project = "
|
| 32 |
-
batch_size =
|
| 33 |
-
base_learning_rate =
|
| 34 |
-
min_learning_rate =
|
| 35 |
num_epochs = 50
|
| 36 |
sample_interval_share = 5
|
| 37 |
cfg_dropout = 0.15
|
|
|
|
| 28 |
|
| 29 |
# --------------------------- Параметры ---------------------------
|
| 30 |
ds_path = "/workspace/sdxs/datasets/mjnj"
|
| 31 |
+
project = "sdxs_sdxl"
|
| 32 |
+
batch_size = 48
|
| 33 |
+
base_learning_rate = 2.7e-5 #2.7e-5
|
| 34 |
+
min_learning_rate = 1e-5 #2.7e-5
|
| 35 |
num_epochs = 50
|
| 36 |
sample_interval_share = 5
|
| 37 |
cfg_dropout = 0.15
|
unet/config.json
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:157591f861026648dd95e0d11cea822457600324907990e5b6ce26a4b16fbb94
|
| 3 |
+
size 1804
|
unet/diffusion_pytorch_model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:aa41354c60970f4b1082f13cb2b3088c2f04c424a31218bbd79d6ce8ab8fda58
|
| 3 |
+
size 6078588464
|
{sdxs_sdxl2 → unet_old}/config.json
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:80410b994cba2961422e6bade67a2a6a41abaf90e6dd89c0226402f45ef113cb
|
| 3 |
+
size 1892
|
{sdxs_sdxl2 → unet_old}/diffusion_pytorch_model.safetensors
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7e5b6db0af234e3191d65afd39653d779a2e49bd2377b3fc4e12c3b68df3fded
|
| 3 |
+
size 3722280432
|