diff --git a/README.md b/README.md index e7f77b6..30595bf 100644 --- a/README.md +++ b/README.md @@ -81,17 +81,17 @@ Then you need to clone our repository into that directory including the extra in **Windows** ``` -pip install aTrain_core@git+https://github.com/JuergenFleiss/aTrain_core.git --extra-index-url https://download.pytorch.org/whl/cu118 +pip install aTrain_core@git+https://github.com/JuergenFleiss/aTrain_core.git@update_fasterwhisper_pyannote --extra-index-url https://download.pytorch.org/whl/cu121 ``` **Debian** ``` -pip3 install aTrain_core@git+https://github.com/JuergenFleiss/aTrain_core.git --extra-index-url https://download.pytorch.org/whl/cu118 +pip3 install aTrain_core@git+https://github.com/JuergenFleiss/aTrain_core.git@update_fasterwhisper_pyannote --extra-index-url https://download.pytorch.org/whl/cu121 ``` 💡 Linux keeps killing collection of torch? Try adding the flag--no-cache-dir **MacOS** ``` -pip3 install aTrain_core@git+https://github.com/JuergenFleiss/aTrain_core.git +pip3 install aTrain_core@git+https://github.com/JuergenFleiss/aTrain_core.git@update_fasterwhisper_pyannote ``` 💡 Note: At the moment, NVIDIA CUDA GPU support is only available for Windows and Debian. We are working on speeding up the transcriptions on MacOS. Hence no need to install the extra index on MacOS. diff --git a/setup.py b/setup.py index b4d7e33..93698d8 100644 --- a/setup.py +++ b/setup.py @@ -4,9 +4,9 @@ system = platform.system() if system in ["Windows","Linux"]: - torch = "torch==2.0.0+cu118" + torch = "torch==2.2.0+cu121" if system == "Darwin": - torch = "torch==2.0.0" + torch = "torch==2.2.0" main_ns = {} ver_path = convert_path('aTrain_core/version.py') @@ -22,17 +22,17 @@ python_requires=">=3.10", install_requires=[ torch, - "torchaudio==2.0.1", - "faster-whisper>=0.8", + "torchaudio==2.2.0", + "faster-whisper==1.0.2", "transformers", "ffmpeg-python>=0.2", "pandas", - "pyannote.audio==3.0.0"], + "pyannote.audio==3.2.0"], packages=find_packages(), include_package_data=True, entry_points={ 'console_scripts': ['aTrain_core = aTrain_core.cli:cli',] } ) - +#