Commit e218f5b
committed
feat: Add Pi0.5 inference support with PaliGemmaWithExpert architecture
Core Model Implementation:
- Add PI0_5_Policy model with 32-dimensional action space support
- Implement discrete state input processing for robotics tasks
- Add PaliGemmaWithExpert backbone with AdaRMSNorm for flow matching
- Support 16-step action prediction with temporal modeling
Inference Configuration:
- Add comprehensive inference configuration templates
- Support both high-level and detailed inference configs
- Include tokenizer and action dimension parameters
- Maintain compatibility with existing Pi0 inference patterns
Technical Details:
- Extended Pi0 architecture for expert-enhanced multimodal reasoning
- Flow matching timestep injection with adaRMS normalization
- Vision-language-action model with discrete state integration
Code Quality:
- Applied Black code formatting
- Fixed trailing whitespace and line endings
- Ensured isort import organization compliance1 parent da4935e commit e218f5b
File tree
4 files changed
+1060
-1
lines changed- examples/pi0_5/conf
- inference
- flagscale/models/pi0
4 files changed
+1060
-1
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
| 19 | + | |
| 20 | + | |
| 21 | + | |
| 22 | + | |
| 23 | + | |
| 24 | + | |
| 25 | + | |
| 26 | + | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
| 19 | + | |
| 20 | + | |
| 21 | + | |
| 22 | + | |
| 23 | + | |
| 24 | + | |
| 25 | + | |
| 26 | + | |
| 27 | + | |
| 28 | + | |
| 29 | + | |
| 30 | + | |
| 31 | + | |
| 32 | + | |
| 33 | + | |
| 34 | + | |
| 35 | + | |
| 36 | + | |
| 37 | + | |
0 commit comments