2512
Browse files- samples/sdxs_08b_192x320_0.jpg +2 -2
- samples/sdxs_08b_256x320_0.jpg +2 -2
- samples/sdxs_08b_320x192_0.jpg +2 -2
- samples/sdxs_08b_320x256_0.jpg +2 -2
- samples/sdxs_08b_320x320_0.jpg +2 -2
- sdxs_08b/diffusion_pytorch_model.safetensors +1 -1
- train.py +2 -2
- train_flux.py +3 -3
samples/sdxs_08b_192x320_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/sdxs_08b_256x320_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/sdxs_08b_320x192_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/sdxs_08b_320x256_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
samples/sdxs_08b_320x320_0.jpg
CHANGED
|
Git LFS Details
|
|
Git LFS Details
|
sdxs_08b/diffusion_pytorch_model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 3376002424
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e06782dffbfbadb10652c76a824c0b8c9f2136722d04be700134846b52a98c41
|
| 3 |
size 3376002424
|
train.py
CHANGED
|
@@ -29,12 +29,12 @@ from transformers import AutoTokenizer, AutoModel
|
|
| 29 |
# --------------------------- Параметры ---------------------------
|
| 30 |
ds_path = "/workspace/sdxs/datasets/mjnj"
|
| 31 |
project = "sdxs_08b"
|
| 32 |
-
batch_size =
|
| 33 |
base_learning_rate = 4e-5 #2.7e-5
|
| 34 |
min_learning_rate = 1e-5 #2.7e-5
|
| 35 |
num_epochs = 15
|
| 36 |
sample_interval_share = 5
|
| 37 |
-
cfg_dropout = 0.
|
| 38 |
max_length = 192
|
| 39 |
use_wandb = True
|
| 40 |
use_comet_ml = False
|
|
|
|
| 29 |
# --------------------------- Параметры ---------------------------
|
| 30 |
ds_path = "/workspace/sdxs/datasets/mjnj"
|
| 31 |
project = "sdxs_08b"
|
| 32 |
+
batch_size = 192
|
| 33 |
base_learning_rate = 4e-5 #2.7e-5
|
| 34 |
min_learning_rate = 1e-5 #2.7e-5
|
| 35 |
num_epochs = 15
|
| 36 |
sample_interval_share = 5
|
| 37 |
+
cfg_dropout = 0.2
|
| 38 |
max_length = 192
|
| 39 |
use_wandb = True
|
| 40 |
use_comet_ml = False
|
train_flux.py
CHANGED
|
@@ -31,10 +31,10 @@ ds_path = "/workspace/sdxs/datasets/mjnj"
|
|
| 31 |
project = "sdxs_flux"
|
| 32 |
batch_size = 256
|
| 33 |
base_learning_rate = 4e-5 #2.7e-5
|
| 34 |
-
min_learning_rate =
|
| 35 |
-
num_epochs =
|
| 36 |
sample_interval_share = 3
|
| 37 |
-
cfg_dropout = 0.
|
| 38 |
max_length = 192
|
| 39 |
use_wandb = True
|
| 40 |
use_comet_ml = False
|
|
|
|
| 31 |
project = "sdxs_flux"
|
| 32 |
batch_size = 256
|
| 33 |
base_learning_rate = 4e-5 #2.7e-5
|
| 34 |
+
min_learning_rate = 1e-5 #2.7e-5
|
| 35 |
+
num_epochs = 15
|
| 36 |
sample_interval_share = 3
|
| 37 |
+
cfg_dropout = 0.25
|
| 38 |
max_length = 192
|
| 39 |
use_wandb = True
|
| 40 |
use_comet_ml = False
|