sdxxs
Browse files- samples/sdxxs_448x768_0.jpg +2 -2
- samples/sdxxs_512x768_0.jpg +2 -2
- samples/sdxxs_576x768_0.jpg +2 -2
- samples/sdxxs_640x768_0.jpg +2 -2
- samples/sdxxs_704x768_0.jpg +2 -2
- samples/sdxxs_768x448_0.jpg +2 -2
- samples/sdxxs_768x512_0.jpg +2 -2
- samples/sdxxs_768x576_0.jpg +2 -2
- samples/sdxxs_768x640_0.jpg +2 -2
- samples/sdxxs_768x704_0.jpg +2 -2
- samples/sdxxs_768x768_0.jpg +2 -2
- sdxxs/config.json +2 -2
- sdxxs/diffusion_pytorch_model.safetensors +2 -2
- train.py +6 -6
- train_sdxxs.py +1 -1
samples/sdxxs_448x768_0.jpg
CHANGED
![]() |
Git LFS Details
|
![]() |
Git LFS Details
|
samples/sdxxs_512x768_0.jpg
CHANGED
![]() |
Git LFS Details
|
![]() |
Git LFS Details
|
samples/sdxxs_576x768_0.jpg
CHANGED
![]() |
Git LFS Details
|
![]() |
Git LFS Details
|
samples/sdxxs_640x768_0.jpg
CHANGED
![]() |
Git LFS Details
|
![]() |
Git LFS Details
|
samples/sdxxs_704x768_0.jpg
CHANGED
![]() |
Git LFS Details
|
![]() |
Git LFS Details
|
samples/sdxxs_768x448_0.jpg
CHANGED
![]() |
Git LFS Details
|
![]() |
Git LFS Details
|
samples/sdxxs_768x512_0.jpg
CHANGED
![]() |
Git LFS Details
|
![]() |
Git LFS Details
|
samples/sdxxs_768x576_0.jpg
CHANGED
![]() |
Git LFS Details
|
![]() |
Git LFS Details
|
samples/sdxxs_768x640_0.jpg
CHANGED
![]() |
Git LFS Details
|
![]() |
Git LFS Details
|
samples/sdxxs_768x704_0.jpg
CHANGED
![]() |
Git LFS Details
|
![]() |
Git LFS Details
|
samples/sdxxs_768x768_0.jpg
CHANGED
![]() |
Git LFS Details
|
![]() |
Git LFS Details
|
sdxxs/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:a2032fabecadeac2a9d886f373defa908bd6b29be62f1c07d31b10aa88316201
|
3 |
+
size 1951
|
sdxxs/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:847244d3abe7215e9aeb575c4ba18fcb3652dfdb82b4ee9bb0e80edd76638c7b
|
3 |
+
size 7010934208
|
train.py
CHANGED
@@ -23,13 +23,13 @@ import bitsandbytes as bnb
|
|
23 |
|
24 |
# --------------------------- Параметры ---------------------------
|
25 |
save_path = "datasets/768" # "datasets/576" #"datasets/576p2" #"datasets/1152p2" #"datasets/576p2" #"datasets/dataset384_temp" #"datasets/dataset384" #"datasets/imagenet-1kk" #"datasets/siski576" #"datasets/siski384" #"datasets/siski64" #"datasets/mnist"
|
26 |
-
batch_size =
|
27 |
-
base_learning_rate = 4e-6 #9.5e-7 #9e-7 #2e-6 #1e-6 #9e-7 #1e-6 #2e-6 #1e-6 #2e-6 #6e-6 #2e-6 #8e-7 #6e-6 #2e-5 #4e-5 #3e-5 #5e-5 #8e-5
|
28 |
min_learning_rate = 2.5e-5 #2e-5
|
29 |
num_epochs = 1 #2 #36 #18
|
30 |
-
project = "sdxxxs"
|
31 |
-
use_wandb =
|
32 |
-
save_model =
|
33 |
limit = 0 #200000 #0
|
34 |
checkpoints_folder = ""
|
35 |
|
@@ -37,7 +37,7 @@ checkpoints_folder = ""
|
|
37 |
n_diffusion_steps = 40
|
38 |
samples_to_generate = 12
|
39 |
guidance_scale = 5
|
40 |
-
sample_interval_share =
|
41 |
|
42 |
# Папки для сохранения результатов
|
43 |
generated_folder = "samples"
|
|
|
23 |
|
24 |
# --------------------------- Параметры ---------------------------
|
25 |
save_path = "datasets/768" # "datasets/576" #"datasets/576p2" #"datasets/1152p2" #"datasets/576p2" #"datasets/dataset384_temp" #"datasets/dataset384" #"datasets/imagenet-1kk" #"datasets/siski576" #"datasets/siski384" #"datasets/siski64" #"datasets/mnist"
|
26 |
+
batch_size = 26 #26 #45 #11 #45 #555 #35 #7
|
27 |
+
base_learning_rate = 2.5e-5 #4e-6 #9.5e-7 #9e-7 #2e-6 #1e-6 #9e-7 #1e-6 #2e-6 #1e-6 #2e-6 #6e-6 #2e-6 #8e-7 #6e-6 #2e-5 #4e-5 #3e-5 #5e-5 #8e-5
|
28 |
min_learning_rate = 2.5e-5 #2e-5
|
29 |
num_epochs = 1 #2 #36 #18
|
30 |
+
project = "sdxxs" #"sdxxxs"
|
31 |
+
use_wandb = True
|
32 |
+
save_model = True
|
33 |
limit = 0 #200000 #0
|
34 |
checkpoints_folder = ""
|
35 |
|
|
|
37 |
n_diffusion_steps = 40
|
38 |
samples_to_generate = 12
|
39 |
guidance_scale = 5
|
40 |
+
sample_interval_share = 25 # samples/save per epoch
|
41 |
|
42 |
# Папки для сохранения результатов
|
43 |
generated_folder = "samples"
|
train_sdxxs.py
CHANGED
@@ -23,7 +23,7 @@ import bitsandbytes as bnb
|
|
23 |
|
24 |
# --------------------------- Параметры ---------------------------
|
25 |
save_path = "datasets/768" # "datasets/576" #"datasets/576p2" #"datasets/1152p2" #"datasets/576p2" #"datasets/dataset384_temp" #"datasets/dataset384" #"datasets/imagenet-1kk" #"datasets/siski576" #"datasets/siski384" #"datasets/siski64" #"datasets/mnist"
|
26 |
-
batch_size =
|
27 |
base_learning_rate = 2.5e-5 #4e-6 #2e-5 #4e-6 #9.5e-7 #9e-7 #2e-6 #1e-6 #9e-7 #1e-6 #2e-6 #1e-6 #2e-6 #6e-6 #2e-6 #8e-7 #6e-6 #2e-5 #4e-5 #3e-5 #5e-5 #8e-5
|
28 |
min_learning_rate = 2.5e-5 #2e-5
|
29 |
num_epochs = 1 #2 #36 #18
|
|
|
23 |
|
24 |
# --------------------------- Параметры ---------------------------
|
25 |
save_path = "datasets/768" # "datasets/576" #"datasets/576p2" #"datasets/1152p2" #"datasets/576p2" #"datasets/dataset384_temp" #"datasets/dataset384" #"datasets/imagenet-1kk" #"datasets/siski576" #"datasets/siski384" #"datasets/siski64" #"datasets/mnist"
|
26 |
+
batch_size = 50 #30 #26 #45 #11 #45 #555 #35 #7
|
27 |
base_learning_rate = 2.5e-5 #4e-6 #2e-5 #4e-6 #9.5e-7 #9e-7 #2e-6 #1e-6 #9e-7 #1e-6 #2e-6 #1e-6 #2e-6 #6e-6 #2e-6 #8e-7 #6e-6 #2e-5 #4e-5 #3e-5 #5e-5 #8e-5
|
28 |
min_learning_rate = 2.5e-5 #2e-5
|
29 |
num_epochs = 1 #2 #36 #18
|