Hugging Face's logo

AiArtLab
/
sdxs

Text-to-Image
Diffusers
Safetensors
Model card Files
xet
Community
recoilme commited on
Commit
d486341
·
1 Parent(s): 14fb16d

2512

Browse files
Files changed (27) hide show
  1. samples/unet_384x768_0.jpg +2 -2
  2. samples/unet_416x768_0.jpg +2 -2
  3. samples/unet_448x768_0.jpg +2 -2
  4. samples/unet_480x768_0.jpg +2 -2
  5. samples/unet_512x768_0.jpg +2 -2
  6. samples/unet_544x768_0.jpg +2 -2
  7. samples/unet_576x768_0.jpg +2 -2
  8. samples/unet_608x768_0.jpg +2 -2
  9. samples/unet_640x768_0.jpg +2 -2
  10. samples/unet_672x768_0.jpg +2 -2
  11. samples/unet_704x768_0.jpg +2 -2
  12. samples/unet_736x768_0.jpg +2 -2
  13. samples/unet_768x384_0.jpg +2 -2
  14. samples/unet_768x416_0.jpg +2 -2
  15. samples/unet_768x448_0.jpg +2 -2
  16. samples/unet_768x480_0.jpg +2 -2
  17. samples/unet_768x512_0.jpg +2 -2
  18. samples/unet_768x544_0.jpg +2 -2
  19. samples/unet_768x576_0.jpg +2 -2
  20. samples/unet_768x608_0.jpg +2 -2
  21. samples/unet_768x640_0.jpg +2 -2
  22. samples/unet_768x672_0.jpg +2 -2
  23. samples/unet_768x704_0.jpg +2 -2
  24. samples/unet_768x736_0.jpg +2 -2
  25. samples/unet_768x768_0.jpg +2 -2
  26. train.py +5 -5
  27. unet/diffusion_pytorch_model.safetensors +1 -1
samples/unet_384x768_0.jpg CHANGED

Git LFS Details

  • SHA256: 4f8fb62e62e0521fd724e47ccd00de9edc179f968a469b0b9774efedda11102e
  • Pointer size: 130 Bytes
  • Size of remote file: 91.4 kB

Git LFS Details

  • SHA256: a609e3e7a06a5163338d9a06575107d5382e1ef32bee2d0200b1f2f7d2cd2a21
  • Pointer size: 130 Bytes
  • Size of remote file: 81.7 kB
samples/unet_416x768_0.jpg CHANGED

Git LFS Details

  • SHA256: 112ee4acdb39f4570b734ebc23352c4fd052cefadf360d8bbef40a43772f8798
  • Pointer size: 130 Bytes
  • Size of remote file: 93.9 kB

Git LFS Details

  • SHA256: 40d672a56d18db7c85013c9023d638fb407116fa64620a7dedf0d9de74d9aaea
  • Pointer size: 130 Bytes
  • Size of remote file: 78.2 kB
samples/unet_448x768_0.jpg CHANGED

Git LFS Details

  • SHA256: aaa7a45cbd7d06d6dbb0a01c01fd76db415999a36f0c657e5df92936814417f0
  • Pointer size: 131 Bytes
  • Size of remote file: 122 kB

Git LFS Details

  • SHA256: 75d8f36fd4a3e55289df7ba8a879b1b56a40e1a05d98a5e326810f1a45661a93
  • Pointer size: 130 Bytes
  • Size of remote file: 71.4 kB
samples/unet_480x768_0.jpg CHANGED

Git LFS Details

  • SHA256: dadc891f210dde52af728522f85828631c8cd56f3464a9c12e083abcd3a7cd68
  • Pointer size: 130 Bytes
  • Size of remote file: 90.9 kB

Git LFS Details

  • SHA256: cbc340360be8ad6d2e43fe5f6605b5fb351f67e1448a00f6b75f60c2807dab29
  • Pointer size: 130 Bytes
  • Size of remote file: 68.4 kB
samples/unet_512x768_0.jpg CHANGED

Git LFS Details

  • SHA256: da4da3d07afa0fc9586ef83a3d57ef2aecf82e95e5bf6a5209fbbb7cf09e1f5f
  • Pointer size: 131 Bytes
  • Size of remote file: 148 kB

Git LFS Details

  • SHA256: 8a215ec09f95f4549a1b663dcc6cb44e0a708f49fa7df0ea904bcd28da5733e2
  • Pointer size: 130 Bytes
  • Size of remote file: 89.3 kB
samples/unet_544x768_0.jpg CHANGED

Git LFS Details

  • SHA256: 2c15c383dcc3e706fd05b93ba26c92a1f9c44175ae6018be6724b82d5ccd1b6b
  • Pointer size: 131 Bytes
  • Size of remote file: 162 kB

Git LFS Details

  • SHA256: 6128896671b828612ed0b1e6eae34062a418c651a73c539f2a53d001230cc5eb
  • Pointer size: 130 Bytes
  • Size of remote file: 80.5 kB
samples/unet_576x768_0.jpg CHANGED

Git LFS Details

  • SHA256: 340ad13619065893b60f1af073e7eddf77467efe66a0fbaef8e820a24812a2ae
  • Pointer size: 131 Bytes
  • Size of remote file: 172 kB

Git LFS Details

  • SHA256: 120f8a9870e12bdf9dcd15203c9e8f26f9054918b753e80e85a69fcdb143753a
  • Pointer size: 131 Bytes
  • Size of remote file: 144 kB
samples/unet_608x768_0.jpg CHANGED

Git LFS Details

  • SHA256: 162b6a728385b2b887e8d5131a9548c11535ee2b8cb02ae751a5e926f9267635
  • Pointer size: 131 Bytes
  • Size of remote file: 108 kB

Git LFS Details

  • SHA256: b7e9d0e31d71af0d39ff4576cbe28ac9dec945c1ca9a46206be8207095ef7974
  • Pointer size: 131 Bytes
  • Size of remote file: 180 kB
samples/unet_640x768_0.jpg CHANGED

Git LFS Details

  • SHA256: 4f239c8276cbe46ccee1f476f199101f757ab334a6c183ce4e0b565958d08b4f
  • Pointer size: 131 Bytes
  • Size of remote file: 136 kB

Git LFS Details

  • SHA256: 83e85c297c483e5e864acf9e62a0ea5b078a2afba7382cf8f24660110fb16757
  • Pointer size: 131 Bytes
  • Size of remote file: 150 kB
samples/unet_672x768_0.jpg CHANGED

Git LFS Details

  • SHA256: ba424e7167e937ed68fb0d6f25acf5fa62563df7d3244396e703528def2df97a
  • Pointer size: 131 Bytes
  • Size of remote file: 105 kB

Git LFS Details

  • SHA256: a892375dce22c56985b7227b106722138fe90662ecccb8f393e3751260b78664
  • Pointer size: 131 Bytes
  • Size of remote file: 151 kB
samples/unet_704x768_0.jpg CHANGED

Git LFS Details

  • SHA256: e2a31dc3a000024c7c83ab8b60eff0bb46a0bec8e1e9a5fed8e0c1a1ef65e6a3
  • Pointer size: 130 Bytes
  • Size of remote file: 48.9 kB

Git LFS Details

  • SHA256: ebf3cf8236d7d8153e934f3bf4ece859ee9029fad1e3cc03ff1afc06f4250a2c
  • Pointer size: 131 Bytes
  • Size of remote file: 232 kB
samples/unet_736x768_0.jpg CHANGED

Git LFS Details

  • SHA256: be68e7d6c9bb4309be8935c0b5fae0090befb8a080944d5d39215b99dc43e4e1
  • Pointer size: 131 Bytes
  • Size of remote file: 148 kB

Git LFS Details

  • SHA256: 7e9c1c1692c1ffd717d93b681f1aaf94b29f094a494c81f4c337002a4a1d4f30
  • Pointer size: 131 Bytes
  • Size of remote file: 223 kB
samples/unet_768x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 01052f22ab8d173d6bc19807f2d7cd7fdc51ab230362af47bf155b5abc687bb0
  • Pointer size: 130 Bytes
  • Size of remote file: 99.4 kB

Git LFS Details

  • SHA256: 2e34c844ce9d34a060a69e6c3ffc53feaaa445006889c60276ba0e719ad49ece
  • Pointer size: 131 Bytes
  • Size of remote file: 154 kB
samples/unet_768x416_0.jpg CHANGED

Git LFS Details

  • SHA256: 61d053082464a2ef4d357b5bd4c5b77a7dd24b277de41304895d98d184cada71
  • Pointer size: 131 Bytes
  • Size of remote file: 133 kB

Git LFS Details

  • SHA256: b055669aaa702d1677d6779a016747b2419d2a2182110e9170d737acb1948b10
  • Pointer size: 130 Bytes
  • Size of remote file: 87.4 kB
samples/unet_768x448_0.jpg CHANGED

Git LFS Details

  • SHA256: 5103b01ac9f1061310fa79ce93ed04dbad6df9dee0ad5a4aa192da72e6acdf1b
  • Pointer size: 130 Bytes
  • Size of remote file: 46.6 kB

Git LFS Details

  • SHA256: bb66f3db155d1b5a6c17435d7bead78050953a3c74a9b409728033044faa6b47
  • Pointer size: 131 Bytes
  • Size of remote file: 202 kB
samples/unet_768x480_0.jpg CHANGED

Git LFS Details

  • SHA256: 9e982e4d050f8f31857281d4dae1a3348cf869f6e3b7fa8cf5430c856f71cc7a
  • Pointer size: 131 Bytes
  • Size of remote file: 126 kB

Git LFS Details

  • SHA256: ffaaa3b5d8065d02095f40c4d020d30f86663648cd879ec256adc31a596a7d6d
  • Pointer size: 131 Bytes
  • Size of remote file: 169 kB
samples/unet_768x512_0.jpg CHANGED

Git LFS Details

  • SHA256: 4d0e1c7839e4bedcfb33df99f681799469f4d6f5990b94d86e5d28702f87477f
  • Pointer size: 131 Bytes
  • Size of remote file: 115 kB

Git LFS Details

  • SHA256: 193531db395599eacc631943a64dfb09fba224d4b42e1aa5c305e4caa68438f9
  • Pointer size: 131 Bytes
  • Size of remote file: 173 kB
samples/unet_768x544_0.jpg CHANGED

Git LFS Details

  • SHA256: e6d09d5c30706d9efc0531e0b1ca4ccd3539b221651c3c59666f1d044713cc18
  • Pointer size: 131 Bytes
  • Size of remote file: 118 kB

Git LFS Details

  • SHA256: 4264e8fb8bf6eae9666914eef106c7bd099ab47221fcbd6ed867a044d81ad246
  • Pointer size: 131 Bytes
  • Size of remote file: 227 kB
samples/unet_768x576_0.jpg CHANGED

Git LFS Details

  • SHA256: 4925b9b8c2064bc92ace8d508b1e922424693ec3bf29f754fb876fd20de4c490
  • Pointer size: 130 Bytes
  • Size of remote file: 14.7 kB

Git LFS Details

  • SHA256: a30593fc17923401171750f7b7d7d375dc8d2e11f26235cf8f3fcdf33d81ad9a
  • Pointer size: 131 Bytes
  • Size of remote file: 206 kB
samples/unet_768x608_0.jpg CHANGED

Git LFS Details

  • SHA256: 61f3468cdd89fe5fb9c212c948b7ad77e40bc9ec749213fa06cdf5f03b6c9f14
  • Pointer size: 131 Bytes
  • Size of remote file: 191 kB

Git LFS Details

  • SHA256: 3bb83dc5c306d5a4848033852910df9fa93763c7d2d75f9ce48df9e089855251
  • Pointer size: 131 Bytes
  • Size of remote file: 122 kB
samples/unet_768x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 51f43adaa45c5220d124abcee8e7886bf983269200fdfafd805dff5710ffeb41
  • Pointer size: 131 Bytes
  • Size of remote file: 202 kB

Git LFS Details

  • SHA256: 532ade4ab519f6b9cdec894f2c75c1b95486574e7e9d695253c120d0d7f13360
  • Pointer size: 131 Bytes
  • Size of remote file: 170 kB
samples/unet_768x672_0.jpg CHANGED

Git LFS Details

  • SHA256: 79f9fec86c9897eca1c51484ffb34af108465638a561c3fed3999b59bf141b04
  • Pointer size: 131 Bytes
  • Size of remote file: 179 kB

Git LFS Details

  • SHA256: 4844056facbe5409dba8d64e7fd00666b9600a5a64a5622af4681162b8a2c281
  • Pointer size: 130 Bytes
  • Size of remote file: 63.3 kB
samples/unet_768x704_0.jpg CHANGED

Git LFS Details

  • SHA256: 03da75605bbd7179f4db6a312c57dc1f01d63a5d80ad3cb5e5ac137f79095544
  • Pointer size: 131 Bytes
  • Size of remote file: 145 kB

Git LFS Details

  • SHA256: f98b6a40b3d8b6dfaa62d379e00be0271d1f7eaef96de3d177b079140f8635c5
  • Pointer size: 131 Bytes
  • Size of remote file: 213 kB
samples/unet_768x736_0.jpg CHANGED

Git LFS Details

  • SHA256: 2fa3ab918cbe76148e4e9675cc3c33870ce546981b9d0d6204155bef7bdc0816
  • Pointer size: 131 Bytes
  • Size of remote file: 137 kB

Git LFS Details

  • SHA256: 4141777de154a2d2f1303b580673d4180d46cf3f0a9260a5e594eac4b523a322
  • Pointer size: 131 Bytes
  • Size of remote file: 125 kB
samples/unet_768x768_0.jpg CHANGED

Git LFS Details

  • SHA256: b09988015005804afb2a0ff23a90b9665affb3d79efad5667b6a8192cdfdc28e
  • Pointer size: 131 Bytes
  • Size of remote file: 236 kB

Git LFS Details

  • SHA256: 895f7b590936a40685d7d2934c3f3786832c009474dffceec21dca5353f03a70
  • Pointer size: 131 Bytes
  • Size of remote file: 198 kB
train.py CHANGED
@@ -30,13 +30,13 @@ from transformers import AutoTokenizer, AutoModel
30
  ds_path = "/workspace/sdxs/datasets/768"
31
  project = "unet"
32
  batch_size = 36
33
- base_learning_rate = 4e-5
34
- min_learning_rate = 2.7e-5
35
  num_epochs = 100
36
  sample_interval_share = 5
37
  max_length = 192
38
- use_wandb = False
39
- use_comet_ml = True
40
  save_model = True
41
  use_decay = True
42
  fbp = False
@@ -53,7 +53,7 @@ torch.backends.cudnn.allow_tf32 = True
53
  dtype = torch.float32
54
  save_barrier = 1.006
55
  warmup_percent = 0.01
56
- percentile_clipping = 97
57
  betta2 = 0.999
58
  eps = 1e-7
59
  clip_grad_norm = 1.0
 
30
  ds_path = "/workspace/sdxs/datasets/768"
31
  project = "unet"
32
  batch_size = 36
33
+ base_learning_rate = 3e-5 #4e-5
34
+ min_learning_rate = 1e-5 #2.7e-5
35
  num_epochs = 100
36
  sample_interval_share = 5
37
  max_length = 192
38
+ use_wandb = True
39
+ use_comet_ml = False
40
  save_model = True
41
  use_decay = True
42
  fbp = False
 
53
  dtype = torch.float32
54
  save_barrier = 1.006
55
  warmup_percent = 0.01
56
+ percentile_clipping = 96 #97
57
  betta2 = 0.999
58
  eps = 1e-7
59
  clip_grad_norm = 1.0
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6cc5185dd74a2f7f37a69da3abeff488cbb497ac87cdcf10ade431e92441777c
3
  size 7444321360
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aba9f6e3b52417efdbb081599870e577ba72416da92d192ecbec48c1958fd8d3
3
  size 7444321360