Hugging Face's logo

AiArtLab
/
sdxs

Text-to-Image
Diffusers
Safetensors
Model card Files
xet
Community
recoilme commited on
Commit
3007b8c
·
1 Parent(s): bff7b6b

ep9

Browse files
Files changed (13) hide show
  1. dataset_fromfolder.py +1 -1
  2. result_grid.jpg +2 -2
  3. samples/unet_192x384_0.jpg +2 -2
  4. samples/unet_256x384_0.jpg +2 -2
  5. samples/unet_320x384_0.jpg +2 -2
  6. samples/unet_384x192_0.jpg +2 -2
  7. samples/unet_384x256_0.jpg +2 -2
  8. samples/unet_384x320_0.jpg +2 -2
  9. samples/unet_384x384_0.jpg +2 -2
  10. test.ipynb +2 -2
  11. train.py +1 -1
  12. unet/diffusion_pytorch_model.fp16.safetensors +1 -1
  13. unet/diffusion_pytorch_model.safetensors +1 -1
dataset_fromfolder.py CHANGED
@@ -18,7 +18,7 @@ from datetime import timedelta
18
  # ---------------- 1️⃣ Настройки ----------------
19
  dtype = torch.float16
20
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
21
- batch_size = 10
22
  min_size = 320 #192
23
  max_size = 576 #384
24
  step = 64
 
18
  # ---------------- 1️⃣ Настройки ----------------
19
  dtype = torch.float16
20
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
21
+ batch_size = 20
22
  min_size = 320 #192
23
  max_size = 576 #384
24
  step = 64
result_grid.jpg CHANGED

Git LFS Details

  • SHA256: 61e4d9c26e8629fc743c1bc1aee9fe6fe7ddb995a8dd77f74f19a77c14011c62
  • Pointer size: 132 Bytes
  • Size of remote file: 6.64 MB

Git LFS Details

  • SHA256: 4cbf1bbbe931782fa4044a599248be5620b1e1f1f87678020b17f592a5d2ac68
  • Pointer size: 132 Bytes
  • Size of remote file: 6.77 MB
samples/unet_192x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 8a9724f03fdf947d57857eb7f64a4553e95b9a65d121fe13e2130c66d545797c
  • Pointer size: 130 Bytes
  • Size of remote file: 34.4 kB

Git LFS Details

  • SHA256: 64615d5d8a1b84e8ceefa70a85c4dfd821f5068f01de3ff9a951f9bfcbd2555e
  • Pointer size: 130 Bytes
  • Size of remote file: 31 kB
samples/unet_256x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 72e91777b3e1f48e72853b11f64b9d0790489a23b47765205e2ca6d1470e491b
  • Pointer size: 130 Bytes
  • Size of remote file: 46.8 kB

Git LFS Details

  • SHA256: 401c7151c2c78caf106095f4182ae394a5628d0618451a37761015e9baafd47c
  • Pointer size: 130 Bytes
  • Size of remote file: 43.6 kB
samples/unet_320x384_0.jpg CHANGED

Git LFS Details

  • SHA256: b6ec59a994be4830bc9925d057f03c0432a91e452052553b6c8a252b1d55a0aa
  • Pointer size: 130 Bytes
  • Size of remote file: 59.2 kB

Git LFS Details

  • SHA256: dc620377a8bc374398eaffc0da88c9b4ef0fd12d1d48a6680481dc4f411005e5
  • Pointer size: 130 Bytes
  • Size of remote file: 55 kB
samples/unet_384x192_0.jpg CHANGED

Git LFS Details

  • SHA256: 20b2bc723c71c85969df95a64dee75e0589e8ed5a0f3888aa58a4e641d6c275f
  • Pointer size: 130 Bytes
  • Size of remote file: 39.1 kB

Git LFS Details

  • SHA256: f13a6060c35f2d43d6099949054b31920cc55f70368504b36971bd02356fc2d4
  • Pointer size: 130 Bytes
  • Size of remote file: 39.8 kB
samples/unet_384x256_0.jpg CHANGED

Git LFS Details

  • SHA256: 7cc0e77c4287c3d6a30318a869fb18107469dadc7720c76abb60771ae2b9168a
  • Pointer size: 130 Bytes
  • Size of remote file: 44.2 kB

Git LFS Details

  • SHA256: 6fbf8ac69a0ce5b17f7fc94e3725a34b77fb75dbcaab22151e986df22ba7f572
  • Pointer size: 130 Bytes
  • Size of remote file: 49.6 kB
samples/unet_384x320_0.jpg CHANGED

Git LFS Details

  • SHA256: 03e33a3c1edd861f5a549b5eadb3270b8e826de3cb4a5bb1c78d27751d535979
  • Pointer size: 130 Bytes
  • Size of remote file: 47.2 kB

Git LFS Details

  • SHA256: a3978a0ae322d0e656c791dcafc6b53b0c6f540b7d4bd3c72213200186dacc6c
  • Pointer size: 130 Bytes
  • Size of remote file: 65.4 kB
samples/unet_384x384_0.jpg CHANGED

Git LFS Details

  • SHA256: b6167aadd46fb9d99994fd7309833f97661682bac095467dada43fcca29a770a
  • Pointer size: 130 Bytes
  • Size of remote file: 42.5 kB

Git LFS Details

  • SHA256: ed8684613a56552c529a83222b87f7516392471d142fae8e6276b6ae38ab02a4
  • Pointer size: 130 Bytes
  • Size of remote file: 49.4 kB
test.ipynb CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6bca9de56c7bdda4a032e2c84d15fd5dfc8108aea08fc3186f8203f428b966f8
3
- size 5148457
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:92d6a1b7034bd2de712560bf42723551e6884a9f3c379855329e50dee422c889
3
+ size 1401708
train.py CHANGED
@@ -27,7 +27,7 @@ import torch.nn.functional as F
27
  # --------------------------- Параметры ---------------------------
28
  ds_path = "datasets/384"
29
  project = "unet"
30
- batch_size = 70
31
  base_learning_rate = 2e-5
32
  min_learning_rate = 8e-6
33
  num_epochs = 10
 
27
  # --------------------------- Параметры ---------------------------
28
  ds_path = "datasets/384"
29
  project = "unet"
30
+ batch_size = 50
31
  base_learning_rate = 2e-5
32
  min_learning_rate = 8e-6
33
  num_epochs = 10
unet/diffusion_pytorch_model.fp16.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:38394b9929db3058a5f3cda105dce8edb2407288a1d989f1e6aedffc759c4364
3
  size 7014306128
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba82fba0a530fa8d0a310e5946b98c08bb5e482aa26c65a74fed55e59b27a99c
3
  size 7014306128
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:38394b9929db3058a5f3cda105dce8edb2407288a1d989f1e6aedffc759c4364
3
  size 7014306128
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba82fba0a530fa8d0a310e5946b98c08bb5e482aa26c65a74fed55e59b27a99c
3
  size 7014306128