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

Fix torchvision

This commit is contained in:
Jacky 2021-01-27 01:24:59 +11:00
parent ab1686a681
commit d1b57d8b97
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@ -5,7 +5,7 @@ import os
from setuptools import setup, find_packages
from pathlib import Path
core_req = ['ftfy', 'regex', 'tqdm', 'torch==1.7.1', 'torchvision']
core_req = ['ftfy', 'regex', 'tqdm', 'torch==1.7.1', 'torchvision==0.8.2']
extras_require={
'cuda': ['cudatoolkit==11.0'],
'dev': ['pytest']
@ -15,7 +15,7 @@ package_data.extend([str(x) for x in list(Path('clip').rglob("*.md"))])
setup(
name='clip_by_openai',
version='0.1.1',
version='0.1.1.2',
author="OpenAI (Packaged by Vector AI)",
author_email="dev@vctr.ai",
description="CLIP by OpenAI",