Fix torchvision
This commit is contained in:
parent
ab1686a681
commit
d1b57d8b97
4
setup.py
4
setup.py
|
@ -5,7 +5,7 @@ import os
|
||||||
from setuptools import setup, find_packages
|
from setuptools import setup, find_packages
|
||||||
from pathlib import Path
|
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={
|
extras_require={
|
||||||
'cuda': ['cudatoolkit==11.0'],
|
'cuda': ['cudatoolkit==11.0'],
|
||||||
'dev': ['pytest']
|
'dev': ['pytest']
|
||||||
|
@ -15,7 +15,7 @@ package_data.extend([str(x) for x in list(Path('clip').rglob("*.md"))])
|
||||||
|
|
||||||
setup(
|
setup(
|
||||||
name='clip_by_openai',
|
name='clip_by_openai',
|
||||||
version='0.1.1',
|
version='0.1.1.2',
|
||||||
author="OpenAI (Packaged by Vector AI)",
|
author="OpenAI (Packaged by Vector AI)",
|
||||||
author_email="dev@vctr.ai",
|
author_email="dev@vctr.ai",
|
||||||
description="CLIP by OpenAI",
|
description="CLIP by OpenAI",
|
||||||
|
|
Loading…
Reference in New Issue