pip install whisperx results in installation of torch >2.0.0

[ymednis]

Description:
For proper functionality, WhisperX requires torch version 2.0.0. However, after running:

pip install whisperx

the installed torch version is 2.6.0—even if torch 2.0.0 was previously installed using:

conda install pytorch==2.0.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia

The issue appears to be due to the following dependency chain:

• whisperx depends on pyannote.audio==3.3.2
• pyannote.audio depends on lightning>=2.0.1
• lightning depends on pytorch-lightning
• pytorch-lightning requires torch>=2.1.0

This chain forces the installation of a torch version higher than 2.0.0.

Steps to Reproduce:

Install torch 2.0.0 with:
conda install pytorch==2.0.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia

Run:
pip install whisperx

Verify that torch is upgraded (e.g., using pip show torch), confirming that version 2.6.0 is installed.

System Information:

lsb_release -a
No LSB modules are available. 
Distributor ID: Ubuntu  
Description:    Ubuntu 22.04.3 LTS  
Release:        22.04  
Codename:       jammy

Expected Behavior:
WhisperX should work with torch 2.0.0 as intended without forcing an upgrade.

Actual Behavior:
Torch is upgraded to 2.6.0 due to the dependency requirements of lightning/pytorch-lightning.

Even in case of downgrading the pyannote.audio to version 3.1.1 and 3.0.0 it still uses lightning which requires torch>=2.1.0

3.0.0.txt
3.1.1.txt