peppaking8 commited on
Commit ·
e8750a5
0
Parent(s):
Publish MiniT2I B16 JAX checkpoint
Browse files- .gitattributes +37 -0
- README.md +23 -0
- checkpoint/_CHECKPOINT_METADATA +1 -0
- checkpoint/_METADATA +0 -0
- checkpoint/d/a4d34c699fe8e74db26177305eddf583 +0 -0
- checkpoint/manifest.ocdbt +0 -0
- checkpoint/ocdbt.process_0/d/41ca058a82aec06f3a16f9ee8cb726ad +0 -0
- checkpoint/ocdbt.process_0/d/5f695f9b5c321ed226f0d031ed980780 +3 -0
- checkpoint/ocdbt.process_0/d/69057cddd92f97ba9184f935dd714d60 +0 -0
- checkpoint/ocdbt.process_0/d/72056308355de2a01a4f559cba891dbd +3 -0
- checkpoint/ocdbt.process_0/d/7bcf9ab7c0dcc0430f6a5ee2c036904f +0 -0
- checkpoint/ocdbt.process_0/d/8d0cd760f37f007089d488e43d857c89 +0 -0
- checkpoint/ocdbt.process_0/d/d3e99110897a15fa47968ce0bb49c51f +0 -0
- checkpoint/ocdbt.process_0/d/f106f13dbdb556c1012f293c58c130f4 +0 -0
- checkpoint/ocdbt.process_0/manifest.ocdbt +0 -0
.gitattributes
ADDED
|
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
checkpoint/ocdbt.process_0/d/5f695f9b5c321ed226f0d031ed980780 filter=lfs diff=lfs merge=lfs -text
|
| 2 |
+
checkpoint/ocdbt.process_0/d/72056308355de2a01a4f559cba891dbd filter=lfs diff=lfs merge=lfs -text
|
| 3 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
| 4 |
+
*.arrow filter=lfs diff=lfs merge=lfs -text
|
| 5 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
| 6 |
+
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
| 7 |
+
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
| 8 |
+
*.ftz filter=lfs diff=lfs merge=lfs -text
|
| 9 |
+
*.gz filter=lfs diff=lfs merge=lfs -text
|
| 10 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
| 11 |
+
*.joblib filter=lfs diff=lfs merge=lfs -text
|
| 12 |
+
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
| 13 |
+
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
| 14 |
+
*.model filter=lfs diff=lfs merge=lfs -text
|
| 15 |
+
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
| 16 |
+
*.npy filter=lfs diff=lfs merge=lfs -text
|
| 17 |
+
*.npz filter=lfs diff=lfs merge=lfs -text
|
| 18 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
| 19 |
+
*.ot filter=lfs diff=lfs merge=lfs -text
|
| 20 |
+
*.parquet filter=lfs diff=lfs merge=lfs -text
|
| 21 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
| 22 |
+
*.pickle filter=lfs diff=lfs merge=lfs -text
|
| 23 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
| 24 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
| 25 |
+
*.pth filter=lfs diff=lfs merge=lfs -text
|
| 26 |
+
*.rar filter=lfs diff=lfs merge=lfs -text
|
| 27 |
+
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
| 28 |
+
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
| 29 |
+
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
| 30 |
+
*.tar filter=lfs diff=lfs merge=lfs -text
|
| 31 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
| 32 |
+
*.tgz filter=lfs diff=lfs merge=lfs -text
|
| 33 |
+
*.wasm filter=lfs diff=lfs merge=lfs -text
|
| 34 |
+
*.xz filter=lfs diff=lfs merge=lfs -text
|
| 35 |
+
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
tags:
|
| 3 |
+
- minit2i
|
| 4 |
+
- minit2i
|
| 5 |
+
- jax
|
| 6 |
+
- checkpoint
|
| 7 |
+
license: other
|
| 8 |
+
private: true
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
# MiniT2I-B/16 JAX Checkpoint
|
| 12 |
+
|
| 13 |
+
Model-only JAX OCDBT checkpoint for **MiniT2I-B/16**.
|
| 14 |
+
|
| 15 |
+
- Architecture: `MMDiT_B_16_txt2`
|
| 16 |
+
- Source training step: `290000`
|
| 17 |
+
- Weights: `ema_params_dict["0.99995"]` from the source checkpoint
|
| 18 |
+
- Contents: `params` only; no optimizer state and no non-EMA training params
|
| 19 |
+
- Diffusers export: `MiniT2I/MiniT2I`
|
| 20 |
+
|
| 21 |
+
The checkpoint is stored under `checkpoint/`.
|
| 22 |
+
|
| 23 |
+
For code and usage, see the JAX release: [PeppaKing8/minit2i-jax](https://github.com/PeppaKing8/minit2i-jax).
|
checkpoint/_CHECKPOINT_METADATA
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"item_handlers": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1780202277991333855, "commit_timestamp_nsecs": 1780202281197423016, "custom_metadata": {}}
|
checkpoint/_METADATA
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
checkpoint/d/a4d34c699fe8e74db26177305eddf583
ADDED
|
Binary file (12.7 kB). View file
|
|
|
checkpoint/manifest.ocdbt
ADDED
|
Binary file (118 Bytes). View file
|
|
|
checkpoint/ocdbt.process_0/d/41ca058a82aec06f3a16f9ee8cb726ad
ADDED
|
Binary file (597 Bytes). View file
|
|
|
checkpoint/ocdbt.process_0/d/5f695f9b5c321ed226f0d031ed980780
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5f082f5481540f89e03129153e5ec2c500227e1765d6c3179bcc29964fc65d2f
|
| 3 |
+
size 747520000
|
checkpoint/ocdbt.process_0/d/69057cddd92f97ba9184f935dd714d60
ADDED
|
Binary file (581 Bytes). View file
|
|
|
checkpoint/ocdbt.process_0/d/72056308355de2a01a4f559cba891dbd
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:45306bb24207ec1c7945de0c1eedebff552c2696e77f7d6fb73bbdeefdb2757f
|
| 3 |
+
size 209637376
|
checkpoint/ocdbt.process_0/d/7bcf9ab7c0dcc0430f6a5ee2c036904f
ADDED
|
Binary file (18.1 kB). View file
|
|
|
checkpoint/ocdbt.process_0/d/8d0cd760f37f007089d488e43d857c89
ADDED
|
Binary file (563 Bytes). View file
|
|
|
checkpoint/ocdbt.process_0/d/d3e99110897a15fa47968ce0bb49c51f
ADDED
|
Binary file (794 Bytes). View file
|
|
|
checkpoint/ocdbt.process_0/d/f106f13dbdb556c1012f293c58c130f4
ADDED
|
Binary file (923 Bytes). View file
|
|
|
checkpoint/ocdbt.process_0/manifest.ocdbt
ADDED
|
Binary file (418 Bytes). View file
|
|
|