-
Notifications
You must be signed in to change notification settings - Fork 13
Expand file tree
/
Copy pathsetup.py
More file actions
67 lines (63 loc) · 1.49 KB
/
Copy pathsetup.py
File metadata and controls
67 lines (63 loc) · 1.49 KB
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
# Copyright (c) 2022-2023 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
#
# Licensed under the NVIDIA Source Code License [see LICENSE for details].
"""
Setup of sam2act
Author: Haoquan Fang
"""
from setuptools import setup, find_packages
requirements = [
"numpy",
"scipy",
"einops",
"pyrender",
"transformers",
"omegaconf",
"natsort",
"cffi",
"pandas",
"tensorflow",
"pyquaternion",
"matplotlib",
"bitsandbytes==0.38.1",
#"triton==2.3.1",
"transforms3d",
"clip @ git+https://github.com/openai/CLIP.git",
"wandb",
"iopath",
"hydra-core==1.3.2",
"pip==23.0"
]
__version__ = "0.0.1"
setup(
name="sam2act",
version=__version__,
description="sam2act",
long_description="",
author="Haoquan Fang",
author_email="[email protected]",
url="",
keywords="robotics,computer vision",
classifiers=[
"Programming Language :: Python",
"Programming Language :: Python :: 3.8",
"Natural Language :: English",
"Topic :: Scientific/Engineering",
],
packages=['sam2act'],
install_requires=requirements,
extras_require={
"xformers": [
"xformers @ git+https://github.com/facebookresearch/xformers.git@main#egg=xformers",
],
"real":
[
"robits[real,audio]>=0.5.1"
]
},
entry_points={
'console_scripts': [
'sam2act-agent = sam2act.real.sam2act_cli:cli',
]
}
)