localhorst/CLIP_on_Tesla_K20Xm
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix installable

Browse Source
This commit is contained in:
Jacky 2021-01-26 17:39:37 +11:00
parent 50d5659d73
commit aba7823739
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
setup.py
View File

@ -5,8 +5,10 @@ import os
from setuptools import setup, find_packages from setuptools import setup, find_packages
core_req = ['ftfy', 'regex', 'tqdm', 'torch==1.7.1', 'torchvision'] core_req = ['ftfy', 'regex', 'tqdm', 'torch==1.7.1', 'torchvision']
extras_require={'cuda': ['cudatoolkit==11.0'], extras_require={
'dev': ['pytest']} 'cuda': ['cudatoolkit==11.0'],
'dev': ['pytest']
}
setup( setup(
name='clip_by_openai', name='clip_by_openai',