Hugging Face's logo

AiArtLab
/
sdxs

Text-to-Image
Diffusers
Safetensors
Model card Files
xet
Community
recoilme commited on
Commit
839ab84
·
1 Parent(s): 5de261d

2511

Browse files
Files changed (14) hide show
  1. samples/unet_320x640_0.jpg +2 -2
  2. samples/unet_384x640_0.jpg +2 -2
  3. samples/unet_448x640_0.jpg +2 -2
  4. samples/unet_512x640_0.jpg +2 -2
  5. samples/unet_576x640_0.jpg +2 -2
  6. samples/unet_640x320_0.jpg +2 -2
  7. samples/unet_640x384_0.jpg +2 -2
  8. samples/unet_640x448_0.jpg +2 -2
  9. samples/unet_640x512_0.jpg +2 -2
  10. samples/unet_640x576_0.jpg +2 -2
  11. samples/unet_640x640_0.jpg +2 -2
  12. src/TRAIN.md +3 -0
  13. train.py +4 -4
  14. unet/diffusion_pytorch_model.safetensors +1 -1
samples/unet_320x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 58b3b014ed854cd63ab8f528ecca62f56bf9bac067a62b4f78f5da977bda9baf
  • Pointer size: 130 Bytes
  • Size of remote file: 36.4 kB

Git LFS Details

  • SHA256: 30dfac1da07d4aed092d8d706752146d50d9f360af93fbb67bc91ffaa4200cd4
  • Pointer size: 130 Bytes
  • Size of remote file: 56.6 kB
samples/unet_384x640_0.jpg CHANGED

Git LFS Details

  • SHA256: ee96fa2480934818fcaef9be1d40747fd7a112692bff316538ce0579793527c5
  • Pointer size: 130 Bytes
  • Size of remote file: 68.3 kB

Git LFS Details

  • SHA256: 2faf20aaf468be88676c4784c823cfb61ba52a6c84adf8f14c499b0d00702aef
  • Pointer size: 131 Bytes
  • Size of remote file: 128 kB
samples/unet_448x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 2af7ea0e646704e54bea751c53a230c9a8fce7d9131c6683046eaf9077cdb2a9
  • Pointer size: 131 Bytes
  • Size of remote file: 123 kB

Git LFS Details

  • SHA256: 0f4879a9ba1cc276d1c774d60919d8b36df342f7725dad5bdc6a7081ffa9198b
  • Pointer size: 131 Bytes
  • Size of remote file: 150 kB
samples/unet_512x640_0.jpg CHANGED

Git LFS Details

  • SHA256: ac65115799f60271c8382c862ca6be754072f667f2e0fbeae8c4be256f1d6a8d
  • Pointer size: 130 Bytes
  • Size of remote file: 81.1 kB

Git LFS Details

  • SHA256: e143e0461d8057ef25c1d6c0c6205a9ea38ac09a597c28a188db5f847f47a4f7
  • Pointer size: 130 Bytes
  • Size of remote file: 93.8 kB
samples/unet_576x640_0.jpg CHANGED

Git LFS Details

  • SHA256: cfa98272f4ef8839410cb95ce92123ce33fbd204410197c195082dfa76c0205b
  • Pointer size: 130 Bytes
  • Size of remote file: 51.4 kB

Git LFS Details

  • SHA256: 9815d11144cbbd850c7f13a5b2a008dfa0d07c0412d66243be577457d4d34afb
  • Pointer size: 131 Bytes
  • Size of remote file: 128 kB
samples/unet_640x320_0.jpg CHANGED

Git LFS Details

  • SHA256: 68ebdaa3ec280abafa8f8057e68d340d168a24c17da2854fc8a7314b443fe170
  • Pointer size: 131 Bytes
  • Size of remote file: 108 kB

Git LFS Details

  • SHA256: a9f17651b9952552c605793f4d08b2240a3765407bf47812ec9b2a314d3bd82a
  • Pointer size: 130 Bytes
  • Size of remote file: 96.1 kB
samples/unet_640x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 1adf9828c62eb0a3b8f970b34ae2a13d4fe4a874f95c71f1b1d7fa6be2326416
  • Pointer size: 130 Bytes
  • Size of remote file: 83.7 kB

Git LFS Details

  • SHA256: 05267ba60030f4b1f7e62cccf24e0797ddbccfdc02337062e121228f055117bb
  • Pointer size: 130 Bytes
  • Size of remote file: 71.4 kB
samples/unet_640x448_0.jpg CHANGED

Git LFS Details

  • SHA256: 1bc262721cf3e142942345f7149e671bb10fb558e5cb76db033d1128c0131f8d
  • Pointer size: 131 Bytes
  • Size of remote file: 154 kB

Git LFS Details

  • SHA256: f3fa1c52a948a993dee380c3e633d9c21e3bade085c2c7cdfb9f4fb8d9311535
  • Pointer size: 130 Bytes
  • Size of remote file: 64.4 kB
samples/unet_640x512_0.jpg CHANGED

Git LFS Details

  • SHA256: 652d5b37367107c125e062db803016a05704c55d83a66660e040cca2bb0b284d
  • Pointer size: 130 Bytes
  • Size of remote file: 90.3 kB

Git LFS Details

  • SHA256: f220b6806464b4155dc0ef6b02a99eaefaab1c7409d5c710036d0207beb1a04f
  • Pointer size: 130 Bytes
  • Size of remote file: 93.4 kB
samples/unet_640x576_0.jpg CHANGED

Git LFS Details

  • SHA256: f2978f17dbcf181500050d1babb6a175d82f25ffbc1fc536aaafb2b1e24478ca
  • Pointer size: 130 Bytes
  • Size of remote file: 88.5 kB

Git LFS Details

  • SHA256: d9c10b3ff9b4c2012b6d1f9c50a8e070e88ea109a20221eb1c2afe3f37f19a34
  • Pointer size: 131 Bytes
  • Size of remote file: 177 kB
samples/unet_640x640_0.jpg CHANGED

Git LFS Details

  • SHA256: 8809a8aa69139e70e2ed6f251cedc9f7a2131dc5f3fc1f8157a8bff8ab5193a7
  • Pointer size: 131 Bytes
  • Size of remote file: 232 kB

Git LFS Details

  • SHA256: 1a26a7492f3255b017ce94d88f0b2d63df46bd0255e82e56610ba0f2e1aba707
  • Pointer size: 131 Bytes
  • Size of remote file: 273 kB
src/TRAIN.md CHANGED
@@ -14,6 +14,9 @@ git config --global credential.helper store
14
 
15
  ```
16
  python -m pip install --upgrade pip
 
 
 
17
  pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130 -U # change
18
  pip install flash-attn --no-build-isolation # optional
19
  ```
 
14
 
15
  ```
16
  python -m pip install --upgrade pip
17
+ # ls /etc/alternatives/cuda -l
18
+ # apt-get install cuda-13-0
19
+ # update-alternatives --config cuda
20
  pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130 -U # change
21
  pip install flash-attn --no-build-isolation # optional
22
  ```
train.py CHANGED
@@ -29,11 +29,11 @@ from collections import deque
29
  ds_path = "/workspace/sdxs/datasets/640"
30
  project = "unet"
31
  batch_size = 48
32
- base_learning_rate = 6e-5
33
- min_learning_rate = 3e-5
34
- num_epochs = 35
35
  # samples/save per epoch
36
- sample_interval_share = 2
37
  use_wandb = True
38
  use_comet_ml = False
39
  save_model = True
 
29
  ds_path = "/workspace/sdxs/datasets/640"
30
  project = "unet"
31
  batch_size = 48
32
+ base_learning_rate = 5e-5
33
+ min_learning_rate = 2e-5
34
+ num_epochs = 50
35
  # samples/save per epoch
36
+ sample_interval_share = 5
37
  use_wandb = True
38
  use_comet_ml = False
39
  save_model = True
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:54dc73bcbf0119ec8f673845a70d7aab5fd7e2e3778594046e0eea5f2f649b8b
3
  size 6184944280
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8cbd20924d2216f6684289e74e4f7c68284378737fc3b992b9addb5cbb2354e7
3
  size 6184944280