Upload folder using huggingface_hub

.gitattributes

@@ -34,3 +34,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 123456789.jpg filter=lfs diff=lfs merge=lfs -text
+samples/sample_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/sample_decoded.jpg filter=lfs diff=lfs merge=lfs -text
+samples/sample_real.jpg filter=lfs diff=lfs merge=lfs -text

train_vae.py

@@ -32,7 +32,7 @@ project            = "vae"
 batch_size         = 4
 base_learning_rate = 6e-6
 min_learning_rate  = 9e-7
-num_epochs         = 50
+num_epochs         = 25
 sample_interval_share = 10
 use_wandb          = True
 save_model         = True
@@ -43,7 +43,7 @@ dtype              = torch.float32
 model_resolution   = 256
 high_resolution    = 512 
 limit              = 0
-save_barrier       = 1.03
+save_barrier       = 1.3
 warmup_percent     = 0.01
 percentile_clipping = 99
 beta2              = 0.997

vae/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8aef462535483be8d283418a62726921310dd8adcd60ee8418ebea5836316627
+oid sha256:be45a8eeae580065f412fcd055f7202bba9d1b09ff0e845ee6408b2fc4b81c42
 size 444559412