2512
Browse files
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:48ad68156cb568ecfd59e413355d721b673d2e16c6ca7bacc452c17499e97dda
|
| 3 |
size 3376002424
|
train.py
CHANGED
|
@@ -31,10 +31,10 @@ 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 =
|
| 35 |
-
num_epochs =
|
| 36 |
-
sample_interval_share =
|
| 37 |
-
cfg_dropout = 0.
|
| 38 |
max_length = 192
|
| 39 |
use_wandb = True
|
| 40 |
use_comet_ml = False
|
|
@@ -96,8 +96,8 @@ lora_alpha = 64
|
|
| 96 |
print("init")
|
| 97 |
|
| 98 |
loss_ratios = {
|
| 99 |
-
"mse": 1.
|
| 100 |
-
"mae": 0.
|
| 101 |
}
|
| 102 |
median_coeff_steps = 256
|
| 103 |
|
|
|
|
| 31 |
project = "sdxs_08b"
|
| 32 |
batch_size = 192
|
| 33 |
base_learning_rate = 4e-5 #2.7e-5
|
| 34 |
+
min_learning_rate = 2e-5 #2.7e-5
|
| 35 |
+
num_epochs = 50
|
| 36 |
+
sample_interval_share = 2
|
| 37 |
+
cfg_dropout = 0.15
|
| 38 |
max_length = 192
|
| 39 |
use_wandb = True
|
| 40 |
use_comet_ml = False
|
|
|
|
| 96 |
print("init")
|
| 97 |
|
| 98 |
loss_ratios = {
|
| 99 |
+
"mse": 1.25,
|
| 100 |
+
"mae": 0.25,
|
| 101 |
}
|
| 102 |
median_coeff_steps = 256
|
| 103 |
|