pepijn223 HF Staff commited on
Commit
f45fb5a
·
verified ·
1 Parent(s): 13e8122

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -5
README.md CHANGED
@@ -1,4 +1,4 @@
1
- # PI0 Pi0 Base (PyTorch, 32-bit floating point)
2
 
3
  This is a PyTorch version of the PI0 pi0_base model, converted from the original JAX/Flax implementation.
4
 
@@ -8,9 +8,6 @@ This is a PyTorch version of the PI0 pi0_base model, converted from the original
8
  - **Model Type**: PI0
9
  - **Domain**: Base model (general purpose)
10
  - **Precision**: 32-bit floating point (fp32)
11
- - **Action Dimension**: 32
12
- - **Action Horizon**: 50
13
- - **Max Token Length**: 48
14
  - **Vision Model**: PaliGemma (gemma_2b)
15
  - **Action Expert**: gemma_300m
16
 
@@ -61,7 +58,6 @@ The model consists of:
61
 
62
  This model was trained on robotics datasets appropriate for its domain:
63
  - **DROID models**: Trained on diverse robot manipulation data
64
- - **ALOHA models**: Trained on bimanual manipulation tasks
65
  - **LIBERO models**: Trained on diverse tabletop manipulation scenarios
66
  - **Base models**: Trained on general robotics datasets
67
 
 
1
+ # π₀ - Base
2
 
3
  This is a PyTorch version of the PI0 pi0_base model, converted from the original JAX/Flax implementation.
4
 
 
8
  - **Model Type**: PI0
9
  - **Domain**: Base model (general purpose)
10
  - **Precision**: 32-bit floating point (fp32)
 
 
 
11
  - **Vision Model**: PaliGemma (gemma_2b)
12
  - **Action Expert**: gemma_300m
13
 
 
58
 
59
  This model was trained on robotics datasets appropriate for its domain:
60
  - **DROID models**: Trained on diverse robot manipulation data
 
61
  - **LIBERO models**: Trained on diverse tabletop manipulation scenarios
62
  - **Base models**: Trained on general robotics datasets
63