54 lines
1.2 KiB
TOML
Executable File
54 lines
1.2 KiB
TOML
Executable File
[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"]
|