第一次完整测例跑完

This commit is contained in:
2026-01-18 00:30:10 +08:00
parent ca15cc593b
commit 25c6fc04db
180 changed files with 29305 additions and 0 deletions

53
pyproject.toml Executable file
View File

@@ -0,0 +1,53 @@
[project]
name = "unifolm_wma"
version = "0.0.1"
description = "UnifoLM-WMA-0"
license = { text = "BSD-3-Clause" }
authors = [
{name="Unitree Embodied AI R&D Team", email="rd_xyc@unitree.com" }
]
requires-python = "==3.10.18"
dependencies = [
"decord==0.6.0",
"einops==0.8.0",
"imageio==2.35.1",
"numpy==1.24.2",
"omegaconf==2.3.0",
"opencv-python==4.10.0.84",
"pandas==2.0.0",
"pillow==9.5.0",
"pytorch-lightning==1.9.3",
"pyyaml==6.0",
"setuptools==65.6.3",
"torch==2.3.1",
"torchvision==0.18.1",
"tqdm==4.66.5",
"transformers==4.40.1",
"moviepy==1.0.3",
"av==12.3.0",
"xformers==0.0.27",
"gradio==4.39.0",
"timm==0.9.10",
"scikit-learn==1.5.1",
"open-clip-torch==2.22.0",
"kornia==0.7.3",
"diffusers==0.30.2",
"termcolor==2.4.0",
"draccus==0.11.5",
"accelerate==1.7.0",
"tensorflow-metadata==1.16.1",
"protobuf==3.20.3",
"datasets==3.6.0",
"tensorflow-graphics==2021.12.3",
"fairscale==0.4.13"
]
[build-system]
requires = ["setuptools>=65.6.3", "wheel"]
build-backend = "setuptools.build_meta"
[tool.setuptools]
package-dir = { "" = "src" }
[tool.setuptools.packages.find]
where = ["src"]