Hugging Face's logo

AiArtLab
/
sdxs-1b

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

2604

Browse files
Files changed (24) hide show
  1. samples/unet_384x704_0.jpg +2 -2
  2. samples/unet_416x704_0.jpg +2 -2
  3. samples/unet_448x704_0.jpg +2 -2
  4. samples/unet_480x704_0.jpg +2 -2
  5. samples/unet_512x704_0.jpg +2 -2
  6. samples/unet_544x704_0.jpg +2 -2
  7. samples/unet_576x704_0.jpg +2 -2
  8. samples/unet_608x704_0.jpg +2 -2
  9. samples/unet_640x704_0.jpg +2 -2
  10. samples/unet_672x704_0.jpg +2 -2
  11. samples/unet_704x384_0.jpg +2 -2
  12. samples/unet_704x416_0.jpg +2 -2
  13. samples/unet_704x448_0.jpg +2 -2
  14. samples/unet_704x480_0.jpg +2 -2
  15. samples/unet_704x512_0.jpg +2 -2
  16. samples/unet_704x544_0.jpg +2 -2
  17. samples/unet_704x576_0.jpg +2 -2
  18. samples/unet_704x608_0.jpg +2 -2
  19. samples/unet_704x640_0.jpg +2 -2
  20. samples/unet_704x672_0.jpg +2 -2
  21. samples/unet_704x704_0.jpg +2 -2
  22. train.py +4 -3
  23. unet/config.json +1 -1
  24. unet/diffusion_pytorch_model.safetensors +2 -2
samples/unet_384x704_0.jpg CHANGED

Git LFS Details

  • SHA256: 6e76dff8feff2579934bc1f52c7d2ef70d7a3786b0067d2f8bf6fc9b870d1812
  • Pointer size: 131 Bytes
  • Size of remote file: 369 kB

Git LFS Details

  • SHA256: 0204a0556ed533013e77cf5f5872e497d10ef0c774c8365df9ecccf2ab1e4733
  • Pointer size: 131 Bytes
  • Size of remote file: 374 kB
samples/unet_416x704_0.jpg CHANGED

Git LFS Details

  • SHA256: 18344ffef956616723b87df5bab70c854052171bfd3ffd13b7db57355db9d2af
  • Pointer size: 131 Bytes
  • Size of remote file: 684 kB

Git LFS Details

  • SHA256: 21011261d8f918f673f7ffc8c7fa3756f076588b7f32558e760602518ef03565
  • Pointer size: 131 Bytes
  • Size of remote file: 420 kB
samples/unet_448x704_0.jpg CHANGED

Git LFS Details

  • SHA256: f5fe9b5faa539fcf88dda5f2df89974a0c236c5ac6986c72088fcfb2d85f7b47
  • Pointer size: 131 Bytes
  • Size of remote file: 252 kB

Git LFS Details

  • SHA256: 13e826b99903019ddf77a984746025e5412aafbf0d2fcb8868cbc7a77d7f9304
  • Pointer size: 131 Bytes
  • Size of remote file: 459 kB
samples/unet_480x704_0.jpg CHANGED

Git LFS Details

  • SHA256: 1c5c54a21c862e099e910a79e5c98f3104f51cab1a56f3062321706d50753d0b
  • Pointer size: 131 Bytes
  • Size of remote file: 359 kB

Git LFS Details

  • SHA256: 8406d9d298b9c0f59cb158f214d3ad51d6003b9a286189ad83946b9b336815ab
  • Pointer size: 131 Bytes
  • Size of remote file: 415 kB
samples/unet_512x704_0.jpg CHANGED

Git LFS Details

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

Git LFS Details

  • SHA256: 2979caa83dc9d53feb32f5e933f788b0982b1e11686f98db5a3adde6c80207f3
  • Pointer size: 131 Bytes
  • Size of remote file: 453 kB
samples/unet_544x704_0.jpg CHANGED

Git LFS Details

  • SHA256: e75b92b675f97e72cbbf257193af4b923854c2582d78ecd09967a1bf3cad1ae9
  • Pointer size: 131 Bytes
  • Size of remote file: 462 kB

Git LFS Details

  • SHA256: ed6c6c4dc4c159b602173bd720dc5c606ce1e321ca433832ac08bea4a22b9d90
  • Pointer size: 131 Bytes
  • Size of remote file: 528 kB
samples/unet_576x704_0.jpg CHANGED

Git LFS Details

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

Git LFS Details

  • SHA256: 797cbaa6b9d6d84382371ec2d7c69882e98efd7ea65bae5683129881ff015e35
  • Pointer size: 131 Bytes
  • Size of remote file: 368 kB
samples/unet_608x704_0.jpg CHANGED

Git LFS Details

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

Git LFS Details

  • SHA256: 569eb16115c11514dcedef081617a1d2a3db3a2fce246105186494cf96dfbd61
  • Pointer size: 131 Bytes
  • Size of remote file: 242 kB
samples/unet_640x704_0.jpg CHANGED

Git LFS Details

  • SHA256: 79da1f39ca3956962e5a66adf49636cc8b9352755ee3c7c8feb3b4f1b6582711
  • Pointer size: 131 Bytes
  • Size of remote file: 858 kB

Git LFS Details

  • SHA256: a8497c37229f38afdce7f9343165a29c2edcab2d4de7fd46d0bc4ef518442e9f
  • Pointer size: 131 Bytes
  • Size of remote file: 359 kB
samples/unet_672x704_0.jpg CHANGED

Git LFS Details

  • SHA256: 546e7f35765c1ba42a1399d1eaa3e69089797591dfd1283b1512b21a4594c4c8
  • Pointer size: 131 Bytes
  • Size of remote file: 339 kB

Git LFS Details

  • SHA256: d2f7befd9ae0013312aacb13c20d619bf6262bf3fda2118a58ab510e87e22f71
  • Pointer size: 131 Bytes
  • Size of remote file: 654 kB
samples/unet_704x384_0.jpg CHANGED

Git LFS Details

  • SHA256: b3e1772fc1a032d0bb7cf3b07b09752e45f47c949797ef8142b3e46ded701e7e
  • Pointer size: 131 Bytes
  • Size of remote file: 633 kB

Git LFS Details

  • SHA256: 68e33f96352f39d09dea445f9823e31ed9d998fe4a841e2ce7071b0504054140
  • Pointer size: 131 Bytes
  • Size of remote file: 499 kB
samples/unet_704x416_0.jpg CHANGED

Git LFS Details

  • SHA256: 3d33c601ffd110bebfc65dd3d2ccd07b1116f080354c4f09a8b246626ff38fbd
  • Pointer size: 131 Bytes
  • Size of remote file: 306 kB

Git LFS Details

  • SHA256: b90cce3e8b117379982f9477e85b736ba6e4dbc839a8c698c6c01d76118bfa4e
  • Pointer size: 131 Bytes
  • Size of remote file: 353 kB
samples/unet_704x448_0.jpg CHANGED

Git LFS Details

  • SHA256: a4f38875075cbf838512b3c725a235d4768c1b1498bb0eb7838fe7618721ff8c
  • Pointer size: 131 Bytes
  • Size of remote file: 423 kB

Git LFS Details

  • SHA256: b6dbeede4cc80aad78c934f3a2fdb4ae8307787b18639eb818e084942344fac7
  • Pointer size: 131 Bytes
  • Size of remote file: 332 kB
samples/unet_704x480_0.jpg CHANGED

Git LFS Details

  • SHA256: 130495b5cbda963333c42aba61dcfe7265833d43481f4b6d25bf86948c0fc7ba
  • Pointer size: 131 Bytes
  • Size of remote file: 573 kB

Git LFS Details

  • SHA256: a5a483dd15b919284584d46d2d442087b4ed444984b80fb189da7487182ecb84
  • Pointer size: 131 Bytes
  • Size of remote file: 354 kB
samples/unet_704x512_0.jpg CHANGED

Git LFS Details

  • SHA256: c371ac28492c5dfe95ca07e09cd08cd77972f89a8aa52083c02d515b093c1fd5
  • Pointer size: 131 Bytes
  • Size of remote file: 176 kB

Git LFS Details

  • SHA256: 7bf0e423fc43d6dfed8e28d135bc38dedc3103f42435a825c092c228a9111ee8
  • Pointer size: 131 Bytes
  • Size of remote file: 548 kB
samples/unet_704x544_0.jpg CHANGED

Git LFS Details

  • SHA256: 94297cbbc07edadd74bbda7a314ceda62f4186c504f45bff7301962e400d82d3
  • Pointer size: 131 Bytes
  • Size of remote file: 682 kB

Git LFS Details

  • SHA256: 8b3c3eaa6bb233adfa61c64592d40b9e81d6789fce57fa1bd06ea163f0040c7f
  • Pointer size: 131 Bytes
  • Size of remote file: 383 kB
samples/unet_704x576_0.jpg CHANGED

Git LFS Details

  • SHA256: 7bdd62ad9ed433e73fb190bb2646056a726d76c177979b288f2aba0f0f0ae422
  • Pointer size: 131 Bytes
  • Size of remote file: 672 kB

Git LFS Details

  • SHA256: ae0f3074ec40b30898e68e78d14b1a9e46f3ffa5d122fb76051cd9c3ff9a72a6
  • Pointer size: 131 Bytes
  • Size of remote file: 508 kB
samples/unet_704x608_0.jpg CHANGED

Git LFS Details

  • SHA256: 95b87ccb2250fea35144f3e4380484255ae0aa5cb530fbbff2e4ce71f8436c16
  • Pointer size: 131 Bytes
  • Size of remote file: 446 kB

Git LFS Details

  • SHA256: 33f79a2a5e03c7308f5b769a4fac7c85047b9e0de9c02d4b8e37f556698e51ef
  • Pointer size: 131 Bytes
  • Size of remote file: 432 kB
samples/unet_704x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 05a092008210ea7f4a8a7ee38c74d9e86e9062b383365da619268f18d2bdb249
  • Pointer size: 131 Bytes
  • Size of remote file: 905 kB

Git LFS Details

  • SHA256: d4b131e7794ae1079169e21229ebf4b2515c5774cac5e60e57df14dae698f39a
  • Pointer size: 131 Bytes
  • Size of remote file: 394 kB
samples/unet_704x672_0.jpg CHANGED

Git LFS Details

  • SHA256: 617ad5f717817e89a947a50e26b4f35d5bed76455444a071123bbc119fded29d
  • Pointer size: 131 Bytes
  • Size of remote file: 798 kB

Git LFS Details

  • SHA256: 635eb3ea43b00b3383d89353a2bcb82463a9b91c565ef5452e7d9d7028edcd67
  • Pointer size: 131 Bytes
  • Size of remote file: 929 kB
samples/unet_704x704_0.jpg CHANGED

Git LFS Details

  • SHA256: 73693f0f76600828189ccc25ba18027a1fa9a1bf96d3641ccb0319c41675023a
  • Pointer size: 131 Bytes
  • Size of remote file: 785 kB

Git LFS Details

  • SHA256: 970af9bd2988d50b189156db6a1d9a60381986617515b34e9175c811f635f065
  • Pointer size: 131 Bytes
  • Size of remote file: 542 kB
train.py CHANGED
@@ -717,8 +717,9 @@ for epoch in range(start_epoch, start_epoch + num_epochs):
717
  noise = torch.randn_like(latents, dtype=latents.dtype)
718
 
719
  # 3. Время t (сэмплим, как и раньше, но чуть сжимаем края)
720
- u = torch.rand(latents.shape[0], device=latents.device, dtype=latents.dtype)
721
- t = u * (1 - 2 * 1e-5) + 1e-5 # Теперь t строго в (0.00001 ... 0.99999)
 
722
  # интерполяция между x0 и шумом
723
  noisy_latents = (1.0 - t.view(-1, 1, 1, 1)) * latents + t.view(-1, 1, 1, 1) * noise
724
  # делаем integer timesteps для UNet
@@ -833,4 +834,4 @@ accelerator.free_memory()
833
  if torch.distributed.is_initialized():
834
  torch.distributed.destroy_process_group()
835
 
836
- print("Готово!")
 
717
  noise = torch.randn_like(latents, dtype=latents.dtype)
718
 
719
  # 3. Время t (сэмплим, как и раньше, но чуть сжимаем края)
720
+ #u = torch.rand(latents.shape[0], device=latents.device, dtype=latents.dtype)
721
+ #t = u * (1 - 2 * 1e-5) + 1e-5 # Теперь t строго в (0.00001 ... 0.99999)
722
+ t = torch.sigmoid(torch.randn(latents.shape[0], device=latents.device, dtype=latents.dtype))
723
  # интерполяция между x0 и шумом
724
  noisy_latents = (1.0 - t.view(-1, 1, 1, 1)) * latents + t.view(-1, 1, 1, 1) * noise
725
  # делаем integer timesteps для UNet
 
834
  if torch.distributed.is_initialized():
835
  torch.distributed.destroy_process_group()
836
 
837
+ print("Готово!")
unet/config.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fbb20721f35fd23f45183d6c2341c319ac059296734c98c786b278c7a42e2f50
3
  size 1879
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e4a3681d28d061b35b033d8cfcf2bd59105bafbc2335c3d5186a3d58cee4fff9
3
  size 1879
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1e70f41443a38e7af9b0115a30585cdf2a19a7449950b527d99ff71fa7160af5
3
- size 3210307232
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:de0e556bd49f281a4d02a92b253e33a8aa88eedfb7af9df2db500153ff9d1460
3
+ size 6420443856