recoilme commited on
Commit
08db690
·
1 Parent(s): b5d0a96
2b/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5b14f729336da3da42b4083c72dc56690c4e0d0dfeb3454d05acdc2cb61319e3
3
  size 7993399544
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5d687a5ffb3a861067456a64fdf99ab51eb2a7ff8e9c98a80a60666b5bd6653
3
  size 7993399544
samples/2b_192x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 08bac15f9e3ad5f4bac1cf71727c31f54c1c1c560f57601c9a68f4edac98c0d5
  • Pointer size: 130 Bytes
  • Size of remote file: 30.8 kB

Git LFS Details

  • SHA256: 521b2567aaf8600b32301468fd658af0165e98c48061ceb40a724bbfadbfd7d4
  • Pointer size: 130 Bytes
  • Size of remote file: 48.5 kB
samples/2b_256x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 429750cab73fae6672ae67e8c070bbf36fdf0f8db2edae49b71f5cfac152fec6
  • Pointer size: 130 Bytes
  • Size of remote file: 33.5 kB

Git LFS Details

  • SHA256: 93f859978406f35bd5ddf29751236c370acfaa9a7c63bf9a11726ea7a3f8cfeb
  • Pointer size: 130 Bytes
  • Size of remote file: 31.5 kB
samples/2b_320x384_0.jpg CHANGED

Git LFS Details

  • SHA256: e5d72e328825a9246880388598bee896e0919966a4b7675d6c7615b7d0b5fe4e
  • Pointer size: 130 Bytes
  • Size of remote file: 51.3 kB

Git LFS Details

  • SHA256: 0592584d92a88608eaa8300ed804c713445b8ab0cbf5c4081c0d45572eb2ef38
  • Pointer size: 130 Bytes
  • Size of remote file: 86 kB
samples/2b_384x192_0.jpg CHANGED

Git LFS Details

  • SHA256: c449cb56803826059b44c54bbcd99286e6c2672511d56666d62105f28d0c1d94
  • Pointer size: 130 Bytes
  • Size of remote file: 32 kB

Git LFS Details

  • SHA256: abda46562847446ed3bdb3a60d0d8636f9be2f34358fc4b7350141153505106e
  • Pointer size: 130 Bytes
  • Size of remote file: 26.2 kB
samples/2b_384x256_0.jpg CHANGED

Git LFS Details

  • SHA256: b4c805a92a0cd354bc6cad79d5c0d15b780fcdf0197561adc54a112d75c108de
  • Pointer size: 130 Bytes
  • Size of remote file: 40.4 kB

Git LFS Details

  • SHA256: b0b24ca0ff576daff1c18553abd76783d76f99a6f68263e6ff4333a54969eba4
  • Pointer size: 130 Bytes
  • Size of remote file: 28.8 kB
samples/2b_384x320_0.jpg CHANGED

Git LFS Details

  • SHA256: d9b0391c848589d1f9fe0114e442434e5b74f7a47c702e0f0d84cda355b76b21
  • Pointer size: 130 Bytes
  • Size of remote file: 76.5 kB

Git LFS Details

  • SHA256: 26ceeb81c8953615b0081b867c4e16323b386db23aeb81017583b48fdf8df747
  • Pointer size: 130 Bytes
  • Size of remote file: 71.3 kB
samples/2b_384x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 31524b5e05f159d2284a06f9ecce3cc4c19bce8c36478626a43a3d52bff8ee09
  • Pointer size: 130 Bytes
  • Size of remote file: 87 kB

Git LFS Details

  • SHA256: 103e264530720dbe9a6499bd87e90e2d9503a16b9946be7889eee49729caa06f
  • Pointer size: 131 Bytes
  • Size of remote file: 105 kB
samples/unet_192x384_0.jpg DELETED

Git LFS Details

  • SHA256: d8d47c33575ca2d269fd10a94218a25849d76e86b435bdc7217d365574365c5a
  • Pointer size: 130 Bytes
  • Size of remote file: 41.8 kB
samples/unet_256x384_0.jpg DELETED

Git LFS Details

  • SHA256: ff6d83b228585a4f63bb7ec4b7207b9de985d8686f98349147975ec68888669f
  • Pointer size: 130 Bytes
  • Size of remote file: 59.8 kB
samples/unet_320x384_0.jpg DELETED

Git LFS Details

  • SHA256: fbeb3036774dc7f824e95f4bfc7b4f77361389b8748c5aa99660c3ebc8487f51
  • Pointer size: 130 Bytes
  • Size of remote file: 62.1 kB
samples/unet_320x576_0.jpg DELETED

Git LFS Details

  • SHA256: aefed30c452f7298b886e3c597499cea3720a9e76d2eb50b3b64da5f80d175da
  • Pointer size: 131 Bytes
  • Size of remote file: 194 kB
samples/unet_384x192_0.jpg DELETED

Git LFS Details

  • SHA256: 3a663ebdef49018f214e6a24ba8b9240970660c79b9299f776982ae16f4ec6ad
  • Pointer size: 130 Bytes
  • Size of remote file: 27 kB
samples/unet_384x256_0.jpg DELETED

Git LFS Details

  • SHA256: ca883f077937f493c41c5887ba603c7037749f79d7889965322d1e3df947241c
  • Pointer size: 130 Bytes
  • Size of remote file: 37.5 kB
samples/unet_384x320_0.jpg DELETED

Git LFS Details

  • SHA256: c288afa636c055a236e2d63a3454cefef6784155d1563a20f8d9a60191eef4ad
  • Pointer size: 130 Bytes
  • Size of remote file: 70.7 kB
samples/unet_384x384_0.jpg DELETED

Git LFS Details

  • SHA256: 1bfc183beb4f38ea2eda561f8fdaf0aaa78d9ebf1b59d4124bd9b503a54ab426
  • Pointer size: 130 Bytes
  • Size of remote file: 21.6 kB
samples/unet_384x576_0.jpg DELETED

Git LFS Details

  • SHA256: 4fd7bd0ca83ff8bc75fd0a4e24466cb45f3eccb56432c097c06eb9f6dc63baf3
  • Pointer size: 129 Bytes
  • Size of remote file: 4.19 kB
samples/unet_448x576_0.jpg DELETED

Git LFS Details

  • SHA256: a856faab239a66016e76aa96ec52d859d9dd02543321eb49d26fc33b7780a17d
  • Pointer size: 131 Bytes
  • Size of remote file: 258 kB
samples/unet_512x576_0.jpg DELETED

Git LFS Details

  • SHA256: a95fe1bb4e7ea7bf19ca35665e10069d4a62cd973358be225b24b129b73e22d5
  • Pointer size: 131 Bytes
  • Size of remote file: 172 kB
samples/unet_576x320_0.jpg DELETED

Git LFS Details

  • SHA256: e94106e73fa83e48d92e843d59cacbc9754ead189fe999a3d4793e2163d7f66c
  • Pointer size: 131 Bytes
  • Size of remote file: 148 kB
samples/unet_576x384_0.jpg DELETED

Git LFS Details

  • SHA256: ec7c7ad9846f65bddd66eeecd9924e88c5df255d0afe565d3c60d26277f852b4
  • Pointer size: 130 Bytes
  • Size of remote file: 39.9 kB
samples/unet_576x448_0.jpg DELETED

Git LFS Details

  • SHA256: d20c87df4c055700d3800e4290935abf21e53fe69a251088b1dc8273cb1b2ff3
  • Pointer size: 131 Bytes
  • Size of remote file: 200 kB
samples/unet_576x512_0.jpg DELETED

Git LFS Details

  • SHA256: f314ecfe9968f4ba5675420c9ed52bda1e3bb579414038e32088ac3aa685e54b
  • Pointer size: 131 Bytes
  • Size of remote file: 275 kB
samples/unet_576x576_0.jpg DELETED

Git LFS Details

  • SHA256: f235d741fc5fc8cf49acec705467762c1a0b415023a7c824546188ba8a0ebcc4
  • Pointer size: 131 Bytes
  • Size of remote file: 295 kB
train.py CHANGED
@@ -30,9 +30,9 @@ project = "2b"
30
  batch_size = 32
31
  base_learning_rate = 8e-5
32
  min_learning_rate = 4e-5
33
- num_epochs = 3
34
  # samples/save per epoch
35
- sample_interval_share = 10
36
  use_wandb = True
37
  save_model = True
38
  use_decay = True
@@ -52,9 +52,9 @@ save_barrier = 1.03
52
  dispersive_temperature=0.5
53
  dispersive_weight= 0.05
54
  percentile_clipping = 95 # 8bit optim
55
- betta2 = 0.99 #0.97
56
- eps = 1e-6 #1e-5
57
- clip_grad_norm = 1. #.5
58
  steps_offset = 1 # Scheduler
59
  limit = 0
60
  checkpoints_folder = ""
 
30
  batch_size = 32
31
  base_learning_rate = 8e-5
32
  min_learning_rate = 4e-5
33
+ num_epochs = 27
34
  # samples/save per epoch
35
+ sample_interval_share = 5
36
  use_wandb = True
37
  save_model = True
38
  use_decay = True
 
52
  dispersive_temperature=0.5
53
  dispersive_weight= 0.05
54
  percentile_clipping = 95 # 8bit optim
55
+ betta2 = 0.97
56
+ eps = 1e-5
57
+ clip_grad_norm = .75
58
  steps_offset = 1 # Scheduler
59
  limit = 0
60
  checkpoints_folder = ""