iamkaikai commited on
Commit
6eb7a4a
1 Parent(s): bca0dc8

End of training

Browse files
README.md CHANGED
@@ -37,12 +37,12 @@ image.save("my_image.png")
37
 
38
  These are the key hyperparameters used during training:
39
 
40
- * Epochs: 7
41
  * Learning rate: 1e-07
42
- * Batch size: 8
43
  * Gradient accumulation steps: 1
44
  * Image resolution: 512
45
  * Mixed-precision: fp16
46
 
47
 
48
- More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/iam-kai-kai/text2image-fine-tune/runs/v2iba1q6).
 
37
 
38
  These are the key hyperparameters used during training:
39
 
40
+ * Epochs: 1
41
  * Learning rate: 1e-07
42
+ * Batch size: 1
43
  * Gradient accumulation steps: 1
44
  * Image resolution: 512
45
  * Mixed-precision: fp16
46
 
47
 
48
+ More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/iam-kai-kai/text2image-fine-tune/runs/277egfk1).
amazing-logos-v2-batch-16.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:151ea15fee54e625f89239e7990e01bdcc5e90ddcccf65bdabe86a4ebc8d3a43
3
+ size 3851921368
unet/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.20.0.dev0",
4
- "_name_or_path": "sd/amazing-logos-v2/checkpoint-60000",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
7
  "addition_embed_type_num_heads": 64,
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.20.0.dev0",
4
+ "_name_or_path": "sd/amazing-logos-v2/checkpoint-100000",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
7
  "addition_embed_type_num_heads": 64,
unet/diffusion_pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0b80595a0f5adc903b552d08e6ca92ed30dbed726733072e1ae710e4343e4864
3
  size 3438375973
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4d253d0938e5d47d96799d97fb0be72f241cc5cebf95464e651d62965538ba1c
3
  size 3438375973