54 lines
1.0 KiB
TOML
54 lines
1.0 KiB
TOML
[build-system]
|
|
requires = ["setuptools>=75", "wheel"]
|
|
build-backend = "setuptools.build_meta"
|
|
|
|
[project]
|
|
name = "alpamayo-qwen05b-validation"
|
|
version = "0.1.0"
|
|
description = "Minimal NVIDIA Alpamayo reproduction with a Qwen2.5-0.5B VLM and 105.6M diffusion expert."
|
|
readme = "README.md"
|
|
license = "Apache-2.0"
|
|
requires-python = ">=3.12,<3.13"
|
|
dependencies = [
|
|
"accelerate==1.12.0",
|
|
"av>=16.0.1",
|
|
"einops>=0.8.1",
|
|
"huggingface-hub==0.36.0",
|
|
"matplotlib>=3.9",
|
|
"numpy>=2.1,<3",
|
|
"pandas>=2.2",
|
|
"physical-ai-av>=0.2.0",
|
|
"pillow>=11",
|
|
"psutil>=6",
|
|
"pyarrow>=18",
|
|
"pyyaml>=6",
|
|
"safetensors>=0.5",
|
|
"scipy>=1.14",
|
|
"torch==2.8.0",
|
|
"torchvision==0.23.0",
|
|
"tqdm>=4.67",
|
|
"transformers==4.57.1",
|
|
]
|
|
|
|
[project.optional-dependencies]
|
|
dev = [
|
|
"pytest>=8.3",
|
|
"ruff>=0.9",
|
|
]
|
|
notebook = [
|
|
"ipykernel>=6.29",
|
|
"jupyterlab>=4.4",
|
|
"nbclient>=0.10",
|
|
"nbformat>=5.10",
|
|
]
|
|
|
|
[tool.setuptools.packages.find]
|
|
where = ["src"]
|
|
|
|
[tool.pytest.ini_options]
|
|
testpaths = ["tests"]
|
|
|
|
[tool.ruff]
|
|
line-length = 100
|
|
extend-exclude = ["references"]
|