Hugging Face's logo

AiArtLab
/
sdxs-1b

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

2601

Browse files
Files changed (25) 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 +7 -2
  25. unet/diffusion_pytorch_model.safetensors +1 -1
media/result_grid.jpg CHANGED

Git LFS Details

  • SHA256: 03df9c44b78d71698075fa14d3ae36f6f09b57adf5ae077b689fa24fe48ad076
  • Pointer size: 132 Bytes
  • Size of remote file: 4.76 MB

Git LFS Details

  • SHA256: 61e5977f22eb0676f71239639f088d8a80ce32fc02e8b6f9c3cc9bcf34c08ea5
  • Pointer size: 132 Bytes
  • Size of remote file: 4.48 MB
samples/unet_320x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 159a012997fe6d529937e2726b1fa681c21c6bd36ab72009eecb4ea6e05a04f8
  • Pointer size: 131 Bytes
  • Size of remote file: 235 kB

Git LFS Details

  • SHA256: 75ae5dcc0fa32a5d038a976806456694c20f63d520e72d6a4d564f344d5dfed1
  • Pointer size: 131 Bytes
  • Size of remote file: 310 kB
samples/unet_352x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 901ca1cf5b044d128bdd2ddc5e7ce9e8f98c06642f91e044b82752be8133a725
  • Pointer size: 131 Bytes
  • Size of remote file: 401 kB

Git LFS Details

  • SHA256: bc836b388b9fb7726daa8ab29f812853105d3c87ed21b036d7a756b36a06c398
  • Pointer size: 131 Bytes
  • Size of remote file: 451 kB
samples/unet_384x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 57660b54e8328ba405ec3560003820f83ee93b4499181a49bd2b097a266b84c0
  • Pointer size: 131 Bytes
  • Size of remote file: 344 kB

Git LFS Details

  • SHA256: 33d8435e2fee6585561472e29144314b8d0aed99e86962343c6e89001be62985
  • Pointer size: 131 Bytes
  • Size of remote file: 501 kB
samples/unet_416x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 6d5a0011bca80ac7b6db5457b50626202809b68e543ad7f3d62efa4d7e52c0bc
  • Pointer size: 131 Bytes
  • Size of remote file: 402 kB

Git LFS Details

  • SHA256: 1ef7ddb7c226820c83e201e964b20ca6725f05ddfb09f0fbedf0812f3551bee0
  • Pointer size: 131 Bytes
  • Size of remote file: 450 kB
samples/unet_448x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 9f70e6919bcacb26aa668ecb3f5d48acc459b9ed15c2980ef96dfc90f32cfe86
  • Pointer size: 131 Bytes
  • Size of remote file: 435 kB

Git LFS Details

  • SHA256: 2072bd7120b29bff33cd36c78d70910f5714fd843763008f7559dd0aafcd169a
  • Pointer size: 131 Bytes
  • Size of remote file: 153 kB
samples/unet_480x640_0.jpg CHANGED

Git LFS Details

  • SHA256: dfe38fc23cebc3fe26167bc3302a3ffe8ee2f92bdeb05c661c06c4426772eac4
  • Pointer size: 131 Bytes
  • Size of remote file: 434 kB

Git LFS Details

  • SHA256: f2ca29ed35b5af2ccdabd763338426027b1162b2810ac5fbaec8b4a3d702cc00
  • Pointer size: 131 Bytes
  • Size of remote file: 596 kB
samples/unet_512x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 630fa7939f55e2bd77164e3327395f34459878b0a9f97ea23ecb81a42846c80c
  • Pointer size: 131 Bytes
  • Size of remote file: 485 kB

Git LFS Details

  • SHA256: 6473e727177f5313b9281d86f992f2d1d648f04680663f9fcccd8adb0380de11
  • Pointer size: 131 Bytes
  • Size of remote file: 872 kB
samples/unet_544x640_0.jpg CHANGED

Git LFS Details

  • SHA256: e84546e56111adc3058664995940d977b49ae2173fef416f827aa19efd5bbe26
  • Pointer size: 131 Bytes
  • Size of remote file: 571 kB

Git LFS Details

  • SHA256: 67c99f7cf40d4b44856e13dc7fd92c20344929e23d2d67cca17d980adb8a88bb
  • Pointer size: 131 Bytes
  • Size of remote file: 734 kB
samples/unet_576x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 28016ff1b12de5c92a94005e7ee0b5f80a04f78aca9d94b64fa9b5dee2f47825
  • Pointer size: 131 Bytes
  • Size of remote file: 418 kB

Git LFS Details

  • SHA256: 578de530ea1201850d7d2f0e6787b9052dd8b1cbe8c15f65347ed0a0620c2db1
  • Pointer size: 131 Bytes
  • Size of remote file: 163 kB
samples/unet_608x640_0.jpg CHANGED

Git LFS Details

  • SHA256: cbcfd6de042760fec25f12385e1962fea85cea276c15969602a7f96225077ec2
  • Pointer size: 131 Bytes
  • Size of remote file: 490 kB

Git LFS Details

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

Git LFS Details

  • SHA256: f309ce41033ce90bef67c61ac426c9e6398a3a9fe3a7ccecf76b72fd92886698
  • Pointer size: 131 Bytes
  • Size of remote file: 301 kB

Git LFS Details

  • SHA256: 1f0e73bec5886820dbb1c78c9efbd68a9c0cd400a35b4b7009280a782bbd2ee2
  • Pointer size: 131 Bytes
  • Size of remote file: 479 kB
samples/unet_640x352_0.jpg CHANGED

Git LFS Details

  • SHA256: 4b32e4ea9eb713d0dcd53f4e42799d19fc91bf41f0d01db19843c03086e6525a
  • Pointer size: 131 Bytes
  • Size of remote file: 308 kB

Git LFS Details

  • SHA256: 37fed1a1fe957fcfd802d7ab79f776a723d1431bc23f851d034e2f37619726e3
  • Pointer size: 131 Bytes
  • Size of remote file: 754 kB
samples/unet_640x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 80744a4325d7fcd044f575fdf32f9b85d820445e7b75eca4d7283bad52a97111
  • Pointer size: 131 Bytes
  • Size of remote file: 197 kB

Git LFS Details

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

Git LFS Details

  • SHA256: fbfe7184f6a8462c7035265b118aeb4ac01ed2ded498df3cdfe22d43530a058d
  • Pointer size: 131 Bytes
  • Size of remote file: 337 kB

Git LFS Details

  • SHA256: 2d0e60b3bd8a26e7ccc35476b61f28c9878134a2bda316b4ea899e46ff2759b2
  • Pointer size: 131 Bytes
  • Size of remote file: 407 kB
samples/unet_640x448_0.jpg CHANGED

Git LFS Details

  • SHA256: 8cae3ce4c6ea0f214027144b65a9e31531049f76789eb2ee597d19a9cd274497
  • Pointer size: 131 Bytes
  • Size of remote file: 300 kB

Git LFS Details

  • SHA256: 0663741dd10f42eba5bdf2655e8bfd498983129062b3ce0ba0df59993bb989c4
  • Pointer size: 131 Bytes
  • Size of remote file: 248 kB
samples/unet_640x480_0.jpg CHANGED

Git LFS Details

  • SHA256: 36af1d7e0a30cde2404834e1b3db4ff5995c4827a346038ce93c628e7db65cb5
  • Pointer size: 131 Bytes
  • Size of remote file: 277 kB

Git LFS Details

  • SHA256: 390ae983436d409575ecb8bc5e7852cf63d6cbd60105b995d00a6278973f06f8
  • Pointer size: 131 Bytes
  • Size of remote file: 912 kB
samples/unet_640x512_0.jpg CHANGED

Git LFS Details

  • SHA256: ede8730afe899800be57038c1ae3cb4a196c4ea5b04cb4d8c05b424891943d4a
  • Pointer size: 131 Bytes
  • Size of remote file: 417 kB

Git LFS Details

  • SHA256: ad5cec4faf8a217ad8311e60c410ce2e5104e5e6eabf9194547ce31d46134ed9
  • Pointer size: 131 Bytes
  • Size of remote file: 489 kB
samples/unet_640x544_0.jpg CHANGED

Git LFS Details

  • SHA256: d5e903417db2e80828541716bf26c331fb836ed2f3e9d87ad1df45fb544f12c8
  • Pointer size: 131 Bytes
  • Size of remote file: 516 kB

Git LFS Details

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

Git LFS Details

  • SHA256: b08927bbd97917a06e49d4714cefcd15786c8cffcebd31a33104741e8ac6ad38
  • Pointer size: 131 Bytes
  • Size of remote file: 387 kB

Git LFS Details

  • SHA256: 2281a3ade4b1a6703223e6c27e64a65e44fd3b1cad90fd290170b0573ee75e0b
  • Pointer size: 131 Bytes
  • Size of remote file: 687 kB
samples/unet_640x608_0.jpg CHANGED

Git LFS Details

  • SHA256: 95a33b6189971c72ac3488552998dc1cacb13ec47de11cc7256ae3e056b52a34
  • Pointer size: 131 Bytes
  • Size of remote file: 715 kB

Git LFS Details

  • SHA256: 761a2e2f3e35372cdabf82a74ba3617d371749f3e92c480400b6c8dfcb35e5d4
  • Pointer size: 131 Bytes
  • Size of remote file: 583 kB
samples/unet_640x640_0.jpg CHANGED

Git LFS Details

  • SHA256: de2f371ed7dabe4046f1b25f2f671c4b8fc26dc859e18f423cc129c02cf70515
  • Pointer size: 131 Bytes
  • Size of remote file: 643 kB

Git LFS Details

  • SHA256: 9346fc4799cbeb2c0632a2c3a66952ba910616a6a964925c7cdd579891beff0f
  • Pointer size: 131 Bytes
  • Size of remote file: 428 kB
test.ipynb CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:389b60c97ef6d7b031d1aed0c8b99f94b3e63e2d245f8fefaf6dac276031f462
3
- size 7056927
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9be98a9c1aeaec6d0c0b93667b408150bea8a38eae7bd0799f3cd8ae4babd532
3
+ size 6799947
train.py CHANGED
@@ -568,9 +568,14 @@ def save_checkpoint(unet, variant=""):
568
  os.path.join(checkpoints_folder, f"{project}"), variant=variant
569
  )
570
  else:
571
- model_to_save.save_pretrained(os.path.join(checkpoints_folder, f"{project}"))
 
 
 
 
 
572
 
573
- unet = unet.to(dtype=dtype)
574
 
575
  # --------------------------- Тренировочный цикл ---------------------------
576
  if accelerator.is_main_process:
 
568
  os.path.join(checkpoints_folder, f"{project}"), variant=variant
569
  )
570
  else:
571
+ #model_to_save.save_pretrained(os.path.join(checkpoints_folder, f"{project}"))
572
+ accelerator.save_model(
573
+ unet,
574
+ os.path.join(checkpoints_folder, f"{project}"),
575
+ safe_serialization=True
576
+ )
577
 
578
+ #unet = unet.to(dtype=dtype) #TODO: wtf???
579
 
580
  # --------------------------- Тренировочный цикл ---------------------------
581
  if accelerator.is_main_process:
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7715153fe14ddf14efcad2fdcf8485b3dca9654f7ff59c9d7d0aa0e91dd1c41f
3
  size 3438444088
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:24b2bdd085395fbcde58dc915502167ce1e494584c3839111f8d7a04292a508a
3
  size 3438444088