Upload folder using huggingface_hub

.gitattributes

@@ -37,3 +37,5 @@ saved_model/**/* 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
+samples/sample_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/sample_2.jpg filter=lfs diff=lfs merge=lfs -text

train_vae.py

@@ -33,7 +33,7 @@ batch_size         = 2
 base_learning_rate = 2e-6
 min_learning_rate  = 7e-7
 num_epochs         = 25
-sample_interval_share = 10
+sample_interval_share = 2
 use_wandb          = True
 save_model         = True
 use_decay          = True
@@ -44,7 +44,7 @@ model_resolution   = 384
 high_resolution    = 768 
 limit              = 0
 save_barrier       = 1.3
-warmup_percent     = 0.01
+warmup_percent     = 0.001
 percentile_clipping = 99
 beta2              = 0.997
 eps                = 1e-8

vae/diffusion_pytorch_model.safetensors

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