Hugging Face's logo

AiArtLab
/
sdxs

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

2107

Browse files
Files changed (9) hide show
  1. 2b/diffusion_pytorch_model.safetensors +1 -1
  2. samples/2b_192x384_0.jpg +2 -2
  3. samples/2b_256x384_0.jpg +2 -2
  4. samples/2b_320x384_0.jpg +2 -2
  5. samples/2b_384x192_0.jpg +2 -2
  6. samples/2b_384x256_0.jpg +2 -2
  7. samples/2b_384x320_0.jpg +2 -2
  8. samples/2b_384x384_0.jpg +2 -2
  9. train.py +3 -3
2b/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5fff20947b4e825d7949deb6c21525b68b1664a56e57d31e5ae7b5548318e271
3
  size 7993399544
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b14f729336da3da42b4083c72dc56690c4e0d0dfeb3454d05acdc2cb61319e3
3
  size 7993399544
samples/2b_192x384_0.jpg CHANGED

Git LFS Details

  • SHA256: be8ca40b94176757ae90ea0e06c9fb201b9db6dfe9c5ee70df544c12dadec89d
  • Pointer size: 130 Bytes
  • Size of remote file: 32.3 kB

Git LFS Details

  • SHA256: 08bac15f9e3ad5f4bac1cf71727c31f54c1c1c560f57601c9a68f4edac98c0d5
  • Pointer size: 130 Bytes
  • Size of remote file: 30.8 kB
samples/2b_256x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 61d778e1e28160e61082cd0a258b38d7afe9158b0c99a9787ea70174603e9bdf
  • Pointer size: 130 Bytes
  • Size of remote file: 49.2 kB

Git LFS Details

  • SHA256: 429750cab73fae6672ae67e8c070bbf36fdf0f8db2edae49b71f5cfac152fec6
  • Pointer size: 130 Bytes
  • Size of remote file: 33.5 kB
samples/2b_320x384_0.jpg CHANGED

Git LFS Details

  • SHA256: ced5093458b89ac812c427f2ae9a9ef0575734535ce84ca598e5b22676206b7c
  • Pointer size: 130 Bytes
  • Size of remote file: 50.2 kB

Git LFS Details

  • SHA256: e5d72e328825a9246880388598bee896e0919966a4b7675d6c7615b7d0b5fe4e
  • Pointer size: 130 Bytes
  • Size of remote file: 51.3 kB
samples/2b_384x192_0.jpg CHANGED

Git LFS Details

  • SHA256: 717274365ac54bd61da342db087ca77250c0eb54d396ede1dc21fc1fa2808cd2
  • Pointer size: 130 Bytes
  • Size of remote file: 31.7 kB

Git LFS Details

  • SHA256: c449cb56803826059b44c54bbcd99286e6c2672511d56666d62105f28d0c1d94
  • Pointer size: 130 Bytes
  • Size of remote file: 32 kB
samples/2b_384x256_0.jpg CHANGED

Git LFS Details

  • SHA256: 9a09c1756cf428a188b3fc309ceb3293b7217d937919cd5a76dac2aeb5b15fee
  • Pointer size: 130 Bytes
  • Size of remote file: 26.8 kB

Git LFS Details

  • SHA256: b4c805a92a0cd354bc6cad79d5c0d15b780fcdf0197561adc54a112d75c108de
  • Pointer size: 130 Bytes
  • Size of remote file: 40.4 kB
samples/2b_384x320_0.jpg CHANGED

Git LFS Details

  • SHA256: e89de22a6acb216fe2935712a3a01513ffb38c3c8f86ef232d96150f6067f136
  • Pointer size: 130 Bytes
  • Size of remote file: 61 kB

Git LFS Details

  • SHA256: d9b0391c848589d1f9fe0114e442434e5b74f7a47c702e0f0d84cda355b76b21
  • Pointer size: 130 Bytes
  • Size of remote file: 76.5 kB
samples/2b_384x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 2123857e84c58763efad53cec96bf2d2a11a858635fddf5a7c82dad3074329bb
  • Pointer size: 130 Bytes
  • Size of remote file: 35.3 kB

Git LFS Details

  • SHA256: 31524b5e05f159d2284a06f9ecce3cc4c19bce8c36478626a43a3d52bff8ee09
  • Pointer size: 130 Bytes
  • Size of remote file: 87 kB
train.py CHANGED
@@ -27,10 +27,10 @@ import torch.nn.functional as F
27
  # --------------------------- Параметры ---------------------------
28
  ds_path = "datasets/384"
29
  project = "2b"
30
- batch_size = 8 #50
31
  base_learning_rate = 8e-5
32
  min_learning_rate = 4e-5
33
- num_epochs = 6
34
  # samples/save per epoch
35
  sample_interval_share = 10
36
  use_wandb = True
@@ -39,7 +39,7 @@ use_decay = True
39
  fbp = False # fused backward pass
40
  optimizer_type = "adam8bit"
41
  torch_compile = False
42
- unet_gradient = False
43
  clip_sample = False #Scheduler
44
  fixed_seed = False
45
  shuffle = True
 
27
  # --------------------------- Параметры ---------------------------
28
  ds_path = "datasets/384"
29
  project = "2b"
30
+ batch_size = 32
31
  base_learning_rate = 8e-5
32
  min_learning_rate = 4e-5
33
+ num_epochs = 3
34
  # samples/save per epoch
35
  sample_interval_share = 10
36
  use_wandb = True
 
39
  fbp = False # fused backward pass
40
  optimizer_type = "adam8bit"
41
  torch_compile = False
42
+ unet_gradient = True
43
  clip_sample = False #Scheduler
44
  fixed_seed = False
45
  shuffle = True