Upload with huggingface_hub

args.json

@@ -15,7 +15,7 @@
   "pad_tokens": false,
   "with_prior_preservation": true,
   "prior_loss_weight": 1.0,
-  "num_class_images": 100,
+  "num_class_images": 200,
   "output_dir": "/content/output",
   "seed": null,
   "resolution": 512,
@@ -23,14 +23,14 @@
   "train_text_encoder": false,
   "train_batch_size": 1,
   "sample_batch_size": 4,
-  "num_train_epochs": 1,
-  "max_train_steps": 100,
+  "num_train_epochs": 3,
+  "max_train_steps": 500,
   "gradient_accumulation_steps": 1,
   "gradient_checkpointing": true,
-  "learning_rate": 5e-06,
+  "learning_rate": 2e-06,
   "scale_lr": false,
   "lr_scheduler": "constant",
-  "lr_warmup_steps": 500,
+  "lr_warmup_steps": 0,
   "use_8bit_adam": true,
   "adam_beta1": 0.9,
   "adam_beta2": 0.999,

model.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:215cd9df074c5cadb6ecb3ee8cba64942d4a9e7cf3682b693652fb7bb48ef1ed
+size 2132791380

unet/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2110974613ee1b78d708b8b1517bc6ec33292fd0be763a62913f58c567cca725
+oid sha256:e94e1523002853e22c235bb51ef1e4457f44bf7b9b8bdcacb6df03983130cee8
 size 3438375973