recoilme commited on
Commit
74ec70f
·
1 Parent(s): 8d9d082
samples/sdxs_08b_192x320_0.jpg CHANGED

Git LFS Details

  • SHA256: c137dc9ae273d5ad50b5bf4775cebc5647a9ce068af9fb84d8780d245999878f
  • Pointer size: 130 Bytes
  • Size of remote file: 34.1 kB

Git LFS Details

  • SHA256: ec530ade3da1bd3e5dd330eac72896f48ea697b2983553d55e8666122a8afa89
  • Pointer size: 130 Bytes
  • Size of remote file: 38.8 kB
samples/sdxs_08b_256x320_0.jpg CHANGED

Git LFS Details

  • SHA256: d5bf5d75e7fa1d41589c11d2293a61b3233d5b9225af356101c322b9afeb6b7c
  • Pointer size: 130 Bytes
  • Size of remote file: 13.2 kB

Git LFS Details

  • SHA256: f42b90152c1717250c210cd4ec4a822217f053c4694c12c4577b835d78a2e070
  • Pointer size: 130 Bytes
  • Size of remote file: 64.7 kB
samples/sdxs_08b_320x192_0.jpg CHANGED

Git LFS Details

  • SHA256: 1e79df0301aec73ea63a248d8820d99dd10330fa4990820ebc641421796bb49f
  • Pointer size: 130 Bytes
  • Size of remote file: 32.4 kB

Git LFS Details

  • SHA256: 53d84e656baea6a56df62111cb37507bde7df4d1cc52dc58cee82dcfcad13313
  • Pointer size: 130 Bytes
  • Size of remote file: 20.7 kB
samples/sdxs_08b_320x256_0.jpg CHANGED

Git LFS Details

  • SHA256: 6420857630c8cc49b6a8827cb1beab3a9a16d46a95af3acc4bc4823a2bbd3b43
  • Pointer size: 130 Bytes
  • Size of remote file: 29 kB

Git LFS Details

  • SHA256: 103fd9be221a51eaf4ca01ee3d8d74bf067d194ce04c10a9390af09bab43c23f
  • Pointer size: 130 Bytes
  • Size of remote file: 56.1 kB
samples/sdxs_08b_320x320_0.jpg CHANGED

Git LFS Details

  • SHA256: fd00d15ae13cf563b9f510e7aa3c0b03d1f9e7bd127c13163fa87f88e4adb55d
  • Pointer size: 130 Bytes
  • Size of remote file: 52.3 kB

Git LFS Details

  • SHA256: af8a359a8289ec6a3da95461a6e2e8325d82a46bebe5e1554b7bd1c36c9f5969
  • Pointer size: 130 Bytes
  • Size of remote file: 58.2 kB
sdxs_08b/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f74021d28806f03df406d8e9e5a9e78d3ecbc3dc4e508a8f594282f04e17a964
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 = 100
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.25
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 = 9e-6 #2.7e-5
35
- num_epochs = 20
36
  sample_interval_share = 3
37
- cfg_dropout = 0.5
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