mirror of
https://github.com/SikongJueluo/ya-vla.git
synced 2025-12-20 14:37:50 +08:00
40 lines
885 B
TOML
40 lines
885 B
TOML
[project]
|
|
name = "ya-vla"
|
|
version = "0.1.0"
|
|
description = "Add your description here"
|
|
readme = "README.md"
|
|
requires-python = ">=3.12"
|
|
dependencies = [
|
|
"accelerate>=1.12.0",
|
|
"mujoco>=3.4.0",
|
|
"pillow>=12.0.0",
|
|
"robosuite>=1.5.1",
|
|
"robosuite-models>=1.0.0",
|
|
"torch>=2.9.1",
|
|
"torchvision>=0.24.1",
|
|
"transformers>=4.57.3",
|
|
]
|
|
|
|
[dependency-groups]
|
|
dev = []
|
|
|
|
[tool.basedpyright]
|
|
exclude = [".venv", ".github", "docs", "tests", ".devcontainer"]
|
|
include = ["src"]
|
|
venvPath = "."
|
|
venv = ".venv"
|
|
typeCheckingMode = "basic"
|
|
|
|
[[tool.uv.index]]
|
|
name = "pytorch-cu130"
|
|
url = "https://download.pytorch.org/whl/cu130"
|
|
explicit = true
|
|
|
|
[tool.uv.sources]
|
|
torch = [
|
|
{ index = "pytorch-cu130", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
|
|
]
|
|
torchvision = [
|
|
{ index = "pytorch-cu130", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
|
|
]
|