-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
30 lines (29 loc) · 848 Bytes
/
setup.py
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
from setuptools import setup, find_packages
setup(
name="eCLIP",
version="0.1.0",
packages=find_packages(),
install_requires=[
"datasets==2.15.0",
"hydra-core==1.3.0",
"numpy==1.26.2",
"pandas==2.1.3",
"peft==0.6.2",
"plotly==5.22.0",
"pytorch-lightning==2.1.2",
"rich==13.7.1",
"scikit-image==0.24.0",
"scikit-learn==1.3.2",
"torch==2.1.1",
"torchvision==0.16.1",
"transformers==4.35.2",
"wandb==0.16.0"
],
author="Yogesh Kumar",
author_email="[email protected]",
description="Improving Contrastive Learning with Expert Annotations",
long_description=open("README.md").read(),
long_description_content_type="text/markdown",
license="Apache",
url="https://github.com/ykumards/eCLIP",
)