Hugging Face's logo

KRadim
/
conditional-medium-unet-tiny-imagenet

Text-to-Image
JAX
English
diffusion-model
flax
flax-nnx
unet
tiny-imagenet
conditional-image-generation
from-scratch
Model card Files
xet
 Community
KRadim commited on
Commit
d1d027f
·
verified ·
1 Parent(s): 57b511f

Upload Conditional MediumUNet model (Orbax checkpoint + native MLIR + compiled StableHLO artifact)

Browse files
Files changed (22) hide show
  1. .gitattributes +6 -0
  2. README.md +1 -1
  3. conditional_medium_unet.mlir +0 -0
  4. conditional_medium_unet.stablehlo +1 -1
  5. jax_checkpoint/_CHECKPOINT_METADATA +1 -1
  6. jax_checkpoint/d/b75c2831f55bf7f6f504450f2e56ef76 +0 -0
  7. jax_checkpoint/manifest.ocdbt +0 -0
  8. jax_checkpoint/ocdbt.process_0/d/0c73d503fbadb80b81e06686c198b3b6 +3 -0
  9. jax_checkpoint/ocdbt.process_0/d/0ff4ec23f2d82958708a43fd76c0fe28 +0 -0
  10. jax_checkpoint/ocdbt.process_0/d/105213d989aba97e2e47b08d639a7bf8 +3 -0
  11. jax_checkpoint/ocdbt.process_0/d/25d8cf8c1eb457b5cb717b4ceb1c1ce2 +0 -0
  12. jax_checkpoint/ocdbt.process_0/d/79cb85e9721611c0a0c1137b02afbaff +3 -0
  13. jax_checkpoint/ocdbt.process_0/manifest.ocdbt +0 -0
  14. model_checkpoint/_CHECKPOINT_METADATA +1 -1
  15. model_checkpoint/d/b75c2831f55bf7f6f504450f2e56ef76 +0 -0
  16. model_checkpoint/manifest.ocdbt +0 -0
  17. model_checkpoint/ocdbt.process_0/d/0c73d503fbadb80b81e06686c198b3b6 +3 -0
  18. model_checkpoint/ocdbt.process_0/d/0ff4ec23f2d82958708a43fd76c0fe28 +0 -0
  19. model_checkpoint/ocdbt.process_0/d/105213d989aba97e2e47b08d639a7bf8 +3 -0
  20. model_checkpoint/ocdbt.process_0/d/25d8cf8c1eb457b5cb717b4ceb1c1ce2 +0 -0
  21. model_checkpoint/ocdbt.process_0/d/79cb85e9721611c0a0c1137b02afbaff +3 -0
  22. model_checkpoint/ocdbt.process_0/manifest.ocdbt +0 -0
.gitattributes CHANGED
@@ -100,3 +100,9 @@ jax_checkpoint/ocdbt.process_0/d/13f372b7c4ebfacbcec46566384443ac filter=lfs dif
100
  jax_checkpoint/ocdbt.process_0/d/67dc079883e08ac29336b8401c88efa5 filter=lfs diff=lfs merge=lfs -text
101
  model_checkpoint/ocdbt.process_0/d/13f372b7c4ebfacbcec46566384443ac filter=lfs diff=lfs merge=lfs -text
102
  model_checkpoint/ocdbt.process_0/d/67dc079883e08ac29336b8401c88efa5 filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
100
  jax_checkpoint/ocdbt.process_0/d/67dc079883e08ac29336b8401c88efa5 filter=lfs diff=lfs merge=lfs -text
101
  model_checkpoint/ocdbt.process_0/d/13f372b7c4ebfacbcec46566384443ac filter=lfs diff=lfs merge=lfs -text
102
  model_checkpoint/ocdbt.process_0/d/67dc079883e08ac29336b8401c88efa5 filter=lfs diff=lfs merge=lfs -text
103
+ jax_checkpoint/ocdbt.process_0/d/0c73d503fbadb80b81e06686c198b3b6 filter=lfs diff=lfs merge=lfs -text
104
+ jax_checkpoint/ocdbt.process_0/d/105213d989aba97e2e47b08d639a7bf8 filter=lfs diff=lfs merge=lfs -text
105
+ jax_checkpoint/ocdbt.process_0/d/79cb85e9721611c0a0c1137b02afbaff filter=lfs diff=lfs merge=lfs -text
106
+ model_checkpoint/ocdbt.process_0/d/0c73d503fbadb80b81e06686c198b3b6 filter=lfs diff=lfs merge=lfs -text
107
+ model_checkpoint/ocdbt.process_0/d/105213d989aba97e2e47b08d639a7bf8 filter=lfs diff=lfs merge=lfs -text
108
+ model_checkpoint/ocdbt.process_0/d/79cb85e9721611c0a0c1137b02afbaff filter=lfs diff=lfs merge=lfs -text
README.md CHANGED
@@ -30,7 +30,7 @@ The model architecture (`ConditionalUNet`) predicts the clean image from a noisy
30
  - **Framework:** JAX / Flax NNX
31
  - **Training data:** Tiny ImageNet (200 classes) — full 100,000-image Tiny ImageNet training set
32
  - **Optimizer:** AdamW with warmup-cosine-decay LR schedule, gradient clipping (global norm 1.0)
33
- - **Epochs:** 560
34
  - **Batch size:** 256
35
 
36
  ## Exported Formats
 
30
  - **Framework:** JAX / Flax NNX
31
  - **Training data:** Tiny ImageNet (200 classes) — full 100,000-image Tiny ImageNet training set
32
  - **Optimizer:** AdamW with warmup-cosine-decay LR schedule, gradient clipping (global norm 1.0)
33
+ - **Epochs:** 600
34
  - **Batch size:** 256
35
 
36
  ## Exported Formats
conditional_medium_unet.mlir CHANGED
The diff for this file is too large to render. See raw diff
 
conditional_medium_unet.stablehlo CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:db0555c14b237972074a36d8f77512473a75d96d23c6d00764e8f1e47533fc73
3
  size 3319980
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7a58717d2e6b0cc5946eba75a2dfa5e74720bf229e42520179e302047086799c
3
  size 3319980
jax_checkpoint/_CHECKPOINT_METADATA CHANGED
@@ -1 +1 @@
1
- {"item_handlers": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler", "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1792492900066583362, "commit_timestamp_nsecs": 1792492900323853126, "custom_metadata": {}}
 
1
+ {"item_handlers": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler", "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1792748369513482595, "commit_timestamp_nsecs": 1792748369778323091, "custom_metadata": {}}
jax_checkpoint/d/b75c2831f55bf7f6f504450f2e56ef76 ADDED
Binary file (12.8 kB). View file
 
jax_checkpoint/manifest.ocdbt CHANGED
Binary files a/jax_checkpoint/manifest.ocdbt and b/jax_checkpoint/manifest.ocdbt differ
 
jax_checkpoint/ocdbt.process_0/d/0c73d503fbadb80b81e06686c198b3b6 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a091600e0d1d78d2758aecbd4ccf4dc7931d36e1191998abfa8e9bddee5c3611
3
+ size 2394244
jax_checkpoint/ocdbt.process_0/d/0ff4ec23f2d82958708a43fd76c0fe28 ADDED
Binary file (346 Bytes). View file
 
jax_checkpoint/ocdbt.process_0/d/105213d989aba97e2e47b08d639a7bf8 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d4371281a8c73ada54ec99fb509e67f8be77b9241a6e18010c537835e153af19
3
+ size 190477
jax_checkpoint/ocdbt.process_0/d/25d8cf8c1eb457b5cb717b4ceb1c1ce2 ADDED
Binary file (877 Bytes). View file
 
jax_checkpoint/ocdbt.process_0/d/79cb85e9721611c0a0c1137b02afbaff ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5d3809ab5ec82990fd8a80d1cf0504d98b433d7b9fc6c14475f6e4cb8cf8776d
3
+ size 482859
jax_checkpoint/ocdbt.process_0/manifest.ocdbt CHANGED
Binary files a/jax_checkpoint/ocdbt.process_0/manifest.ocdbt and b/jax_checkpoint/ocdbt.process_0/manifest.ocdbt differ
 
model_checkpoint/_CHECKPOINT_METADATA CHANGED
@@ -1 +1 @@
1
- {"item_handlers": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler", "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1792492900066583362, "commit_timestamp_nsecs": 1792492900323853126, "custom_metadata": {}}
 
1
+ {"item_handlers": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler", "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1792748369513482595, "commit_timestamp_nsecs": 1792748369778323091, "custom_metadata": {}}
model_checkpoint/d/b75c2831f55bf7f6f504450f2e56ef76 ADDED
Binary file (12.8 kB). View file
 
model_checkpoint/manifest.ocdbt CHANGED
Binary files a/model_checkpoint/manifest.ocdbt and b/model_checkpoint/manifest.ocdbt differ
 
model_checkpoint/ocdbt.process_0/d/0c73d503fbadb80b81e06686c198b3b6 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a091600e0d1d78d2758aecbd4ccf4dc7931d36e1191998abfa8e9bddee5c3611
3
+ size 2394244
model_checkpoint/ocdbt.process_0/d/0ff4ec23f2d82958708a43fd76c0fe28 ADDED
Binary file (346 Bytes). View file
 
model_checkpoint/ocdbt.process_0/d/105213d989aba97e2e47b08d639a7bf8 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d4371281a8c73ada54ec99fb509e67f8be77b9241a6e18010c537835e153af19
3
+ size 190477
model_checkpoint/ocdbt.process_0/d/25d8cf8c1eb457b5cb717b4ceb1c1ce2 ADDED
Binary file (877 Bytes). View file
 
model_checkpoint/ocdbt.process_0/d/79cb85e9721611c0a0c1137b02afbaff ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5d3809ab5ec82990fd8a80d1cf0504d98b433d7b9fc6c14475f6e4cb8cf8776d
3
+ size 482859
model_checkpoint/ocdbt.process_0/manifest.ocdbt CHANGED
Binary files a/model_checkpoint/ocdbt.process_0/manifest.ocdbt and b/model_checkpoint/ocdbt.process_0/manifest.ocdbt differ