-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsetup.py
More file actions
33 lines (29 loc) · 872 Bytes
/
setup.py
File metadata and controls
33 lines (29 loc) · 872 Bytes
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
from setuptools import setup
def main():
setup(
name="drivefusion",
version="0.1.0",
author="OmarSamirz",
author_email="omarsamir1300@gmail.com",
description="DriveFusion Code",
long_description="DriveFusion",
long_description_content_type="text/markdown",
keywords=["AI", "LLM", "DLM", "VLM", "DriveFusion"],
license="Apache 2.0 License",
url="https://github.com/DriveFusion/drivefusion",
packages=["drivefusion"],
python_requires=">=3.9.0",
install_requires=[
"transformers==4.52.4",
"torch",
"torchvision",
"torchaudio",
"qwen-vl-utils",
"accelerate",
"wheel",
"llmcompressor",
"deepdiff",
],
)
if __name__ == "__main__":
main()