VG-LLM/setup.py at main · TUe-Ray/VG-LLM · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
import os
from setuptools import setup, find_packages

setup(
    name="vgllm",
    version="0.1.0",
    packages=find_packages("src"),
    package_dir={"": "src"},
    install_requires=[
        "torch==2.5.1",
        "torchvision==0.20.1",
        "transformers==4.50.0",
        "deepspeed==0.16.4",
        "flash_attn==2.7.4.post1",
        "triton==3.1.0",
        "accelerate==1.4.0",
        "torchcodec==0.2",
        "black==24.1.0",
        "isort==5.13.2",
        "datasets==3.6.0",
        "evaluate>=0.4.0",
        "httpx==0.25.0",
        "jsonlines",
        "numexpr",
        "numpy==1.26.4",
        "peft>=0.2.0",
        "pybind11>=2.6.2",
        "pytablewriter",
        "sacrebleu>=1.5.0",
        "scikit-learn>=0.24.1",
        "sqlitedict==2.1.0",
        "timm",
        "einops",
        "ftfy",
        "openai",
        "opencv-python-headless",
        "av",
        "hf_transfer",
        "nltk",
        "sentencepiece==0.1.99",
        "yt-dlp",
        "pycocoevalcap",
        "tqdm-multiprocess",
        "transformers-stream-generator",
        "zstandard",
        "pillow",
        "pyyaml",
        "sympy",
        "mpmath",
        "Jinja2",
        "openpyxl",
        "loguru",
        "hf_transfer",
        "tenacity==8.3.0",
        "wandb>=0.16.0",
        "tiktoken",
        "pre-commit",
        "pydantic",
        "packaging",
        "decord",
        "zss",
        "protobuf==3.20",
        "qwen_vl_utils",
        "open3d===0.19.0",
        "spicy==0.16.0",
        "terminaltables",
        "pytorch3d"
    ],
    author="Duo Zheng, Shijia Huang, Yanyang Li, Liwei Wang",
    author_email="dzheng23@link.cuhk.edu.hk",
    description="Official PyTorch implementation for \"Learning from Videos for 3D World: Enhancing MLLMs with 3D Vision Geometry Priors\"",
    long_description=open("README.md").read() if os.path.exists("README.md") else "",
    long_description_content_type="text/markdown",
    classifiers=[
        "Programming Language :: Python :: 3",
        "Operating System :: OS Independent",
    ],
    python_requires="==3.10.*",
)