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

Git LFS Details

  • SHA256: ec530ade3da1bd3e5dd330eac72896f48ea697b2983553d55e8666122a8afa89
  • Pointer size: 130 Bytes
  • Size of remote file: 38.8 kB

Git LFS Details

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

Git LFS Details

  • SHA256: f42b90152c1717250c210cd4ec4a822217f053c4694c12c4577b835d78a2e070
  • Pointer size: 130 Bytes
  • Size of remote file: 64.7 kB

Git LFS Details

  • SHA256: 38a91d95dc8c0f7549323125a89a87fdde49b18c8232681fb322e5de2c661a42
  • Pointer size: 130 Bytes
  • Size of remote file: 65.8 kB
samples/sdxs_08b_320x192_0.jpg CHANGED

Git LFS Details

  • SHA256: 53d84e656baea6a56df62111cb37507bde7df4d1cc52dc58cee82dcfcad13313
  • Pointer size: 130 Bytes
  • Size of remote file: 20.7 kB

Git LFS Details

  • SHA256: c69e0c25e0d0591acc442a0ab1e3ad8c55f87c73aef919777c498adc3a54b645
  • Pointer size: 130 Bytes
  • Size of remote file: 41.4 kB
samples/sdxs_08b_320x256_0.jpg CHANGED

Git LFS Details

  • SHA256: 103fd9be221a51eaf4ca01ee3d8d74bf067d194ce04c10a9390af09bab43c23f
  • Pointer size: 130 Bytes
  • Size of remote file: 56.1 kB

Git LFS Details

  • SHA256: 31c78a3a04ee0ad30783106425639d8886eb78ff4a4d93e6cdbf8a0de8554f44
  • Pointer size: 130 Bytes
  • Size of remote file: 44.5 kB
samples/sdxs_08b_320x320_0.jpg CHANGED

Git LFS Details

  • SHA256: af8a359a8289ec6a3da95461a6e2e8325d82a46bebe5e1554b7bd1c36c9f5969
  • Pointer size: 130 Bytes
  • Size of remote file: 58.2 kB

Git LFS Details

  • SHA256: 0636b505baba0e5f168f80a83e5e703e83d9637bfca541b6cb5b3d7ed9f56b6b
  • Pointer size: 130 Bytes
  • Size of remote file: 17.7 kB
sdxs_08b/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e06782dffbfbadb10652c76a824c0b8c9f2136722d04be700134846b52a98c41
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 = 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
@@ -96,8 +96,8 @@ lora_alpha = 64
96
  print("init")
97
 
98
  loss_ratios = {
99
- "mse": 1.5,
100
- "mae": 0.5,
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