configured setup.py to handle pytorch

This commit is contained in:
Jaikinator
2023-08-25 14:30:06 +02:00
parent 9fbe850b88
commit e499c98703
+4 -1
View File
@@ -32,10 +32,13 @@ if __name__ == "__main__":
open(os.path.join(os.path.dirname(__file__), "requirements.txt")) open(os.path.join(os.path.dirname(__file__), "requirements.txt"))
) )
], ],
dependency_links=[
'https://download.pytorch.org/whl/cu113',
],
url= github_url, url= github_url,
license='', license='',
author='Jacob Schmieder', author='Jacob Schmieder',
author_email='', author_email='Jacob.Schmieder@dbfz.de',
description='Transcription tool for audio files based on Whisper and Pyannote', description='Transcription tool for audio files based on Whisper and Pyannote',
entry_points={'console_scripts': entry_points={'console_scripts':
['autotranscript = autotranscript.autotranscript:cli']} ['autotranscript = autotranscript.autotranscript:cli']}