diff --git a/requirements.txt b/requirements.txt index 053e612..7d50360 100644 --- a/requirements.txt +++ b/requirements.txt @@ -17,4 +17,4 @@ tqdm yapf einops imgaug -git+https://github.com/openai/CLIP.git \ No newline at end of file +clip @ git+https://github.com/openai/CLIP.git \ No newline at end of file diff --git a/setup.py b/setup.py index 7605b32..63e774c 100644 --- a/setup.py +++ b/setup.py @@ -76,13 +76,7 @@ def get_version(): def get_requirements(filename='requirements.txt'): here = os.path.dirname(os.path.realpath(__file__)) with open(os.path.join(here, filename), 'r') as f: - requires = [] - for line in f.readlines(): - line = line.replace('\n', '') - if 'http' in line and 'CLIP' in line: - requires.append(f'clip @ {line}') - else: - requires.append(line) + requires = [line.replace('\n', '') for line in f.readlines()] return requires