Initial pi0 minimal validation

2026-07-11 20:04:18 +08:00
commit 4fe0099f3f
18 changed files with 1611 additions and 0 deletions
@@ -0,0 +1,35 @@
+[build-system]
+requires = ["setuptools>=69", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "pi0-minimal-validation"
+version = "0.1.0"
+description = "A tiny educational pi0-style VLA flow-matching overfit experiment"
+readme = "README.md"
+requires-python = ">=3.10"
+license = { text = "MIT" }
+dependencies = [
+  "matplotlib>=3.8",
+  "numpy>=1.26",
+  "Pillow>=10.0",
+  "PyYAML>=6.0",
+  "torch>=2.2",
+]
+
+[project.optional-dependencies]
+dev = ["pytest>=8.0"]
+
+[project.scripts]
+pi0mini-train = "pi0mini.train:main"
+pi0mini-infer = "pi0mini.infer:main"
+
+[tool.setuptools]
+package-dir = { "" = "src" }
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+addopts = "-ra"