From d5ba15cf46efb6941e9910a237a5bdcff3eb4d4b Mon Sep 17 00:00:00 2001 From: Jacky Date: Tue, 26 Jan 2021 17:25:39 +1100 Subject: [PATCH] Fix torch installation --- setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index aa8b74a..844b368 100644 --- a/setup.py +++ b/setup.py @@ -4,11 +4,11 @@ from setuptools import setup, find_packages import os -core_req = ['ftfy', 'regex', 'tqdm', 'pytorch==1.7.1', 'torchvision'] +core_req = ['ftfy', 'regex', 'tqdm', 'torch==1.7.1', 'torchvision'] extra_requires={'cuda': ['cudatoolkit==11.0']} setup( - name=name, + name='clip_by_openai', version='0.1.0', author="OpenAI", author_email="dev@vctr.ai",