Hugging Face's logo

AiArtLab
/
sdxs-1b

Text-to-Image
Diffusers
Safetensors
Model card Files
xet
 Community
2
recoilme commited on
Commit
ce7235d
·
1 Parent(s): 2e1cd99

2601

Browse files
Files changed (24) hide show
  1. media/result_grid.jpg +2 -2
  2. samples/unet_320x640_0.jpg +2 -2
  3. samples/unet_352x640_0.jpg +2 -2
  4. samples/unet_384x640_0.jpg +2 -2
  5. samples/unet_416x640_0.jpg +2 -2
  6. samples/unet_448x640_0.jpg +2 -2
  7. samples/unet_480x640_0.jpg +2 -2
  8. samples/unet_512x640_0.jpg +2 -2
  9. samples/unet_544x640_0.jpg +2 -2
  10. samples/unet_576x640_0.jpg +2 -2
  11. samples/unet_608x640_0.jpg +2 -2
  12. samples/unet_640x320_0.jpg +2 -2
  13. samples/unet_640x352_0.jpg +2 -2
  14. samples/unet_640x384_0.jpg +2 -2
  15. samples/unet_640x416_0.jpg +2 -2
  16. samples/unet_640x448_0.jpg +2 -2
  17. samples/unet_640x480_0.jpg +2 -2
  18. samples/unet_640x512_0.jpg +2 -2
  19. samples/unet_640x544_0.jpg +2 -2
  20. samples/unet_640x576_0.jpg +2 -2
  21. samples/unet_640x608_0.jpg +2 -2
  22. samples/unet_640x640_0.jpg +2 -2
  23. test.ipynb +2 -2
  24. train.py +6 -6
media/result_grid.jpg CHANGED

Git LFS Details

  • SHA256: 35d914697468b9b3ff42850a498481b90ac578c221fc51fa2e3b59185c0e38c5
  • Pointer size: 132 Bytes
  • Size of remote file: 4.49 MB

Git LFS Details

  • SHA256: a6db2611bde57a03b712b63ff59f147cce167916604bc6cdfa218004def2d88a
  • Pointer size: 132 Bytes
  • Size of remote file: 4.52 MB
samples/unet_320x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 2c7eab3541bfeb07418deb5fe7337e23ebcaa7c43d69f20e30b777d8d1abff93
  • Pointer size: 131 Bytes
  • Size of remote file: 300 kB

Git LFS Details

  • SHA256: 8cadbade32a5851a82c51472c5475ee954cd9108701287db704816c97c8b9ada
  • Pointer size: 131 Bytes
  • Size of remote file: 315 kB
samples/unet_352x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 1cb68b3e5d0caefccba537ef7a12157fe800248190edce70cfd066eb7afc6406
  • Pointer size: 131 Bytes
  • Size of remote file: 312 kB

Git LFS Details

  • SHA256: 47ae860eb440178763306a36e29a0114d59bbae29eddc0a724d68afc5ed334c4
  • Pointer size: 131 Bytes
  • Size of remote file: 397 kB
samples/unet_384x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 35e26ca1d330374a2d5c8f37b43e80538d58612f8a0125ceda5cf5bfd7e0cf6d
  • Pointer size: 131 Bytes
  • Size of remote file: 141 kB

Git LFS Details

  • SHA256: e6fcfbe8d0cbc3eb298d3f90b3c0a063ad90fceec8312717fd9d37ff5f3e5ae0
  • Pointer size: 131 Bytes
  • Size of remote file: 107 kB
samples/unet_416x640_0.jpg CHANGED

Git LFS Details

  • SHA256: dbd8c6e095c36690e0e632b562aff0ee1962f212c6913000e148952a4c0b9226
  • Pointer size: 131 Bytes
  • Size of remote file: 238 kB

Git LFS Details

  • SHA256: 70fd93a4028640b9ad61f3de0c3b3e69c26626e6f3c565c898ff35505bab442b
  • Pointer size: 131 Bytes
  • Size of remote file: 138 kB
samples/unet_448x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 53439df0e6e9fdde949061b7390788d36a8f06dec0647daef0185df14bfc0557
  • Pointer size: 131 Bytes
  • Size of remote file: 348 kB

Git LFS Details

  • SHA256: 67b896eb75053fa411609d985aec458234fd24cf0dc9475ea087fe5cf0a2f2a3
  • Pointer size: 131 Bytes
  • Size of remote file: 416 kB
samples/unet_480x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 4ff08e23f3737e881f8dce57505b7e773a500e69d7024f5416c17485a78ab5c7
  • Pointer size: 131 Bytes
  • Size of remote file: 401 kB

Git LFS Details

  • SHA256: 071e203df51181d7fed7a55024dfcb2c48abf214a206dc76d8cbeca9d84757eb
  • Pointer size: 131 Bytes
  • Size of remote file: 415 kB
samples/unet_512x640_0.jpg CHANGED

Git LFS Details

  • SHA256: b27fe10a1cb765f8cddf4452469c294aaf1bf79d293e6e1675142a4786d9d1e7
  • Pointer size: 131 Bytes
  • Size of remote file: 370 kB

Git LFS Details

  • SHA256: 9c9f629c329db18dcdbebc972e796ee3372bf57764bc039576755daa312bc22e
  • Pointer size: 131 Bytes
  • Size of remote file: 372 kB
samples/unet_544x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 3bab5628689d2b7e3878312a17e3e7e92f47ee5b722d84ed980e27a9a165affc
  • Pointer size: 131 Bytes
  • Size of remote file: 145 kB

Git LFS Details

  • SHA256: 0a4b80d842aaa2eeed31e82c5a9eca90e9ce84d9ec8b78c7d2f5c4d1c9231220
  • Pointer size: 131 Bytes
  • Size of remote file: 155 kB
samples/unet_576x640_0.jpg CHANGED

Git LFS Details

  • SHA256: fc39a7239dca8de44ac4507943c786e3f3b5ef2d835720a378bd6b65981c8268
  • Pointer size: 131 Bytes
  • Size of remote file: 426 kB

Git LFS Details

  • SHA256: e4b545f899895aee753b4017a8d2c753497191c10312db3047ac7dacbedd880d
  • Pointer size: 131 Bytes
  • Size of remote file: 435 kB
samples/unet_608x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 71d250aea8d00fd45143fd39a8108160b81842e3f6eb630c371b0af4de91ace4
  • Pointer size: 131 Bytes
  • Size of remote file: 453 kB

Git LFS Details

  • SHA256: c2d02edf3df70e4aa77c27c37def9576979bc16d2be28ed3539a1eb3eb030042
  • Pointer size: 131 Bytes
  • Size of remote file: 557 kB
samples/unet_640x320_0.jpg CHANGED

Git LFS Details

  • SHA256: adca86649ddda898f277302a4533aae98e93bc4cac3c31b00d5333af9d9c23f7
  • Pointer size: 131 Bytes
  • Size of remote file: 153 kB

Git LFS Details

  • SHA256: 7ac2154610c9aed2d6375f807e029a574ad28ccecf58812ce0d7e6ce4fd0331b
  • Pointer size: 131 Bytes
  • Size of remote file: 123 kB
samples/unet_640x352_0.jpg CHANGED

Git LFS Details

  • SHA256: d5f6d56d9cb2689a3b2bb291b10a184c3ad1752a5409b089d043bf411c0111f2
  • Pointer size: 131 Bytes
  • Size of remote file: 206 kB

Git LFS Details

  • SHA256: 34914586c597dae9e7426274b5ed15181f1aeae4f2eae3d902142348dc1d71e9
  • Pointer size: 130 Bytes
  • Size of remote file: 77.1 kB
samples/unet_640x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 652f7d4f16d6f6988bb3e551dd426df24e66971e53446c8dcb348907acf9a007
  • Pointer size: 131 Bytes
  • Size of remote file: 666 kB

Git LFS Details

  • SHA256: cd266fcf20202cdae2cd97adfd9854833a853efc9af8e14ef517bc98fb9a49f6
  • Pointer size: 131 Bytes
  • Size of remote file: 683 kB
samples/unet_640x416_0.jpg CHANGED

Git LFS Details

  • SHA256: 9bf0ec16d7703c86562c688c0cbf29e4a3e67a528308a8fca65bb43109d1c140
  • Pointer size: 131 Bytes
  • Size of remote file: 117 kB

Git LFS Details

  • SHA256: d4f1110ac657ab135a799351db9c006fe91d2af7de78d37aab8559d43fc50c7a
  • Pointer size: 130 Bytes
  • Size of remote file: 45.3 kB
samples/unet_640x448_0.jpg CHANGED

Git LFS Details

  • SHA256: fe3a617bdecce806b39ba3d15ac70c67491499e3b6985d5e5606fe723f1ee3f6
  • Pointer size: 131 Bytes
  • Size of remote file: 247 kB

Git LFS Details

  • SHA256: 8d4ab92b3f6a997c10a8ee8fac9a576c0562cfe9a4ec95553e1eabb4227cd57c
  • Pointer size: 131 Bytes
  • Size of remote file: 260 kB
samples/unet_640x480_0.jpg CHANGED

Git LFS Details

  • SHA256: b5bd8f1342157a8723689887ae970cf8079807b120322433c97b361e4a28ef60
  • Pointer size: 131 Bytes
  • Size of remote file: 347 kB

Git LFS Details

  • SHA256: e1cb36984cfe33e1ff1273e2a5e6c7648903e7aaf89067ba313ea29b6ddfaf2b
  • Pointer size: 131 Bytes
  • Size of remote file: 252 kB
samples/unet_640x512_0.jpg CHANGED

Git LFS Details

  • SHA256: a8aba600f2cbca8186a14743202c24ad617916a96e9b940360cbc05041095a9b
  • Pointer size: 131 Bytes
  • Size of remote file: 408 kB

Git LFS Details

  • SHA256: 93e0691132f151d790a00f01d4d040bfd3315ac73a7f9b6ed650024c95ea1e6e
  • Pointer size: 131 Bytes
  • Size of remote file: 520 kB
samples/unet_640x544_0.jpg CHANGED

Git LFS Details

  • SHA256: 570381e6503cd6470566b167078d3f403c3cf40df7a7a561a593fc60652ed632
  • Pointer size: 131 Bytes
  • Size of remote file: 693 kB

Git LFS Details

  • SHA256: c1570350801da01270d735287039b5b4f983fab56af065eae7b0219294fb1059
  • Pointer size: 131 Bytes
  • Size of remote file: 781 kB
samples/unet_640x576_0.jpg CHANGED

Git LFS Details

  • SHA256: 2a75ad5f37c5a471331c4db0e7c835d514ede72807063fc494d453ed4b164050
  • Pointer size: 131 Bytes
  • Size of remote file: 352 kB

Git LFS Details

  • SHA256: 4f33b142e7bb1fadde12f4e39a99bcdc5cd54b901a7bfdd7f1c9e727118f3e19
  • Pointer size: 131 Bytes
  • Size of remote file: 345 kB
samples/unet_640x608_0.jpg CHANGED

Git LFS Details

  • SHA256: f1b959539953655d2c8d00bf7820f9361502f7eabcd47f2d9fd42ca536570f31
  • Pointer size: 131 Bytes
  • Size of remote file: 200 kB

Git LFS Details

  • SHA256: 954038fe4821f8e2a6867943348e39e58d2a792f312c48bddf89c467292bdf95
  • Pointer size: 131 Bytes
  • Size of remote file: 221 kB
samples/unet_640x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 85c1d867f7c29d72c1a98d6f746fe488e6f1842b5d544d5102a5957707442b04
  • Pointer size: 131 Bytes
  • Size of remote file: 247 kB

Git LFS Details

  • SHA256: 786f42a9ddafb1e2d2436d0abe1ca65c09ebf6d10060edbc27baead9b0acb1d0
  • Pointer size: 131 Bytes
  • Size of remote file: 365 kB
test.ipynb CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:572d17c1a60eab5ed75f42f859316930b3daae9bc5945302b4192c929e455d5c
3
- size 4952320
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:560ef25a440535d32333b084043c7d8e5fb729370118ea3d2e576181fdf1ad26
3
+ size 127336
train.py CHANGED
@@ -29,16 +29,16 @@ from transformers import AutoTokenizer, AutoModel
29
  # --------------------------- Параметры ---------------------------
30
  ds_path = "/workspace/sdxs-08b/datasets/d123_640_sd15"
31
  project = "unet"
32
- batch_size = 10
33
- base_learning_rate = 4e-5
34
- min_learning_rate = 2e-5
35
  num_epochs = 20
36
  sample_interval_share = 5
37
  cfg_dropout = 0.10
38
  max_length = 248
39
  use_wandb = False
40
- use_comet_ml = True
41
- save_model = True
42
  use_decay = True
43
  fbp = False
44
  optimizer_type = "adam8bit"
@@ -51,7 +51,7 @@ comet_ml_api_key = "Agctp26mbqnoYrrlvQuKSTk6r"
51
  comet_ml_workspace = "recoilme"
52
  torch.backends.cuda.matmul.allow_tf32 = True
53
  torch.backends.cudnn.allow_tf32 = True
54
- torch.backends.cuda.enable_mem_efficient_sdp(False)
55
  dtype = torch.float32
56
  save_barrier = 1.05
57
  warmup_percent = 0.03
 
29
  # --------------------------- Параметры ---------------------------
30
  ds_path = "/workspace/sdxs-08b/datasets/d123_640_sd15"
31
  project = "unet"
32
+ batch_size = 50
33
+ base_learning_rate = 4e-6
34
+ min_learning_rate = 2e-6
35
  num_epochs = 20
36
  sample_interval_share = 5
37
  cfg_dropout = 0.10
38
  max_length = 248
39
  use_wandb = False
40
+ use_comet_ml = False
41
+ save_model = False
42
  use_decay = True
43
  fbp = False
44
  optimizer_type = "adam8bit"
 
51
  comet_ml_workspace = "recoilme"
52
  torch.backends.cuda.matmul.allow_tf32 = True
53
  torch.backends.cudnn.allow_tf32 = True
54
+ torch.backends.cuda.enable_mem_efficient_sdp(True)
55
  dtype = torch.float32
56
  save_barrier = 1.05
57
  warmup_percent = 0.03