From e499c987038d3f137fd8f1ed3fecb2444b79a45e Mon Sep 17 00:00:00 2001 From: Jaikinator Date: Fri, 25 Aug 2023 14:30:06 +0200 Subject: [PATCH] configured setup.py to handle pytorch --- setup.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index e7da608..bf6a912 100644 --- a/setup.py +++ b/setup.py @@ -32,10 +32,13 @@ if __name__ == "__main__": open(os.path.join(os.path.dirname(__file__), "requirements.txt")) ) ], + dependency_links=[ + 'https://download.pytorch.org/whl/cu113', + ], url= github_url, license='', author='Jacob Schmieder', - author_email='', + author_email='Jacob.Schmieder@dbfz.de', description='Transcription tool for audio files based on Whisper and Pyannote', entry_points={'console_scripts': ['autotranscript = autotranscript.autotranscript:cli']}