Hugging Face's logo

YiYiXu
/
kandinsky_prior_pokemon

Text-to-Image
Diffusers
Safetensors
KandinskyV22PriorPipeline
kandinsky
Model card Files Community
YiYiXu commited on
Commit
3c610d6
•
1 Parent(s): 993f625

End of training

Browse files
Files changed (17) hide show
  1. README.md +1 -1
  2. checkpoint-1000/optimizer.bin +3 -0
  3. checkpoint-1000/prior/config.json +20 -0
  4. checkpoint-1000/prior/diffusion_pytorch_model.safetensors +3 -0
  5. checkpoint-1000/random_states_0.pkl +3 -0
  6. checkpoint-1000/scaler.pt +3 -0
  7. checkpoint-1000/scheduler.bin +3 -0
  8. checkpoint-500/optimizer.bin +3 -0
  9. checkpoint-500/prior/config.json +20 -0
  10. checkpoint-500/prior/diffusion_pytorch_model.safetensors +3 -0
  11. checkpoint-500/random_states_0.pkl +3 -0
  12. checkpoint-500/scaler.pt +3 -0
  13. checkpoint-500/scheduler.bin +3 -0
  14. image_encoder/config.json +1 -1
  15. prior/diffusion_pytorch_model.safetensors +1 -1
  16. text_encoder/config.json +1 -1
  17. val_imgs_grid.png +0 -0
README.md CHANGED
@@ -46,4 +46,4 @@ These are the key hyperparameters used during training:
46
  * Mixed-precision: fp16
47
 
48
 
49
- More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/yiyixu/text2image-fine-tune/runs/90ljjfwe).
 
46
  * Mixed-precision: fp16
47
 
48
 
49
+ More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/yiyixu/text2image-fine-tune/runs/rmf7pvcm).
checkpoint-1000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0f674e5e0b0f65f8dd3547b07d7b768670f8eb66b91808f12690a61c5513a570
3
+ size 8210072757
checkpoint-1000/prior/config.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "PriorTransformer",
3
+ "_diffusers_version": "0.21.0.dev0",
4
+ "_name_or_path": "kandinsky-community/kandinsky-2-2-prior",
5
+ "added_emb_type": "prd",
6
+ "additional_embeddings": 4,
7
+ "attention_head_dim": 64,
8
+ "clip_embed_dim": null,
9
+ "dropout": 0.0,
10
+ "embedding_dim": 1280,
11
+ "embedding_proj_dim": null,
12
+ "embedding_proj_norm_type": null,
13
+ "encoder_hid_proj_type": "linear",
14
+ "norm_in_type": null,
15
+ "num_attention_heads": 32,
16
+ "num_embeddings": 77,
17
+ "num_layers": 20,
18
+ "time_embed_act_fn": "silu",
19
+ "time_embed_dim": null
20
+ }
checkpoint-1000/prior/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:09fb7abd4a2c97d6ad5e4c156cad5031c32749c4d5a86f792039f6e40baea459
3
+ size 4104940968
checkpoint-1000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4921469ee17d35368be2171c5345255478044e55cf225962338702279b2b7f3
3
+ size 14663
checkpoint-1000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68cff80b680ddf6e7abbef98b5f336b97f9b5963e2209307f639383870e8cc71
3
+ size 557
checkpoint-1000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f34912fa092355c5a950e4967ea28cbb3556238a7857baa5999d228cde0ae4c3
3
+ size 563
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e1899552691a690755fecd8571254a1bdd0212847f16b2eb552f06a076c3458
3
+ size 8210072757
checkpoint-500/prior/config.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "PriorTransformer",
3
+ "_diffusers_version": "0.21.0.dev0",
4
+ "_name_or_path": "kandinsky-community/kandinsky-2-2-prior",
5
+ "added_emb_type": "prd",
6
+ "additional_embeddings": 4,
7
+ "attention_head_dim": 64,
8
+ "clip_embed_dim": null,
9
+ "dropout": 0.0,
10
+ "embedding_dim": 1280,
11
+ "embedding_proj_dim": null,
12
+ "embedding_proj_norm_type": null,
13
+ "encoder_hid_proj_type": "linear",
14
+ "norm_in_type": null,
15
+ "num_attention_heads": 32,
16
+ "num_embeddings": 77,
17
+ "num_layers": 20,
18
+ "time_embed_act_fn": "silu",
19
+ "time_embed_dim": null
20
+ }
checkpoint-500/prior/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b85ffbfa020f5ee5e9e9af4ea5a36a6e94afb793bd73fcc9ade31e1fb57aa31b
3
+ size 4104940968
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7734921530583b09c7d79239e43a28d6b8d7893b1c80c8d4e3c71d7b9039ed23
3
+ size 14663
checkpoint-500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3f196a54202bb4ba1220e8c59f42f9cda0702d68ea83147d814c2fb2f36b8f2
3
+ size 557
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab20ead000e6ae60e9a32d69fb1fb976d10dba2a065da680293a8202053a1975
3
+ size 563
image_encoder/config.json CHANGED
@@ -19,5 +19,5 @@
19
  "patch_size": 14,
20
  "projection_dim": 1280,
21
  "torch_dtype": "float16",
22
- "transformers_version": "4.33.0"
23
  }
 
19
  "patch_size": 14,
20
  "projection_dim": 1280,
21
  "torch_dtype": "float16",
22
+ "transformers_version": "4.33.1"
23
  }
prior/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d15ab9af57d6dc140dd3bbc69470be7964a1f87726082c8046ea0961d76f5d3f
3
  size 4104940968
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:09fb7abd4a2c97d6ad5e4c156cad5031c32749c4d5a86f792039f6e40baea459
3
  size 4104940968
text_encoder/config.json CHANGED
@@ -20,6 +20,6 @@
20
  "pad_token_id": 1,
21
  "projection_dim": 1280,
22
  "torch_dtype": "float16",
23
- "transformers_version": "4.33.0",
24
  "vocab_size": 49408
25
  }
 
20
  "pad_token_id": 1,
21
  "projection_dim": 1280,
22
  "torch_dtype": "float16",
23
+ "transformers_version": "4.33.1",
24
  "vocab_size": 49408
25
  }
val_imgs_grid.png CHANGED