From aba78237398e85d5efb51d79c44fff3499e6620f Mon Sep 17 00:00:00 2001 From: Jacky Date: Tue, 26 Jan 2021 17:39:37 +1100 Subject: [PATCH] Fix installable --- setup.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 8cd66ce..40e680a 100644 --- a/setup.py +++ b/setup.py @@ -5,8 +5,10 @@ import os from setuptools import setup, find_packages core_req = ['ftfy', 'regex', 'tqdm', 'torch==1.7.1', 'torchvision'] -extras_require={'cuda': ['cudatoolkit==11.0'], - 'dev': ['pytest']} +extras_require={ + 'cuda': ['cudatoolkit==11.0'], + 'dev': ['pytest'] +} setup( name='clip_by_openai',