fix: Update PyTorch/torchaudio to 2.1.0 for SpeechBrain 1.0 diarization compatibility
- Update torch from 2.0.1 to 2.1.0 - Update torchaudio from 2.0.2 to 2.1.0 (fixes 'NoneType' object has no attribute 'eval' error) - Update torchvision from 0.15.2 to 0.16.0 - Update pytorch-lightning from 2.0.6 to 2.1.0 - Add explicit speechbrain==1.0.0 dependency - Update transformers and tokenizers for compatibility - Update protobuf version constraint Fixes diarization error: speechbrain.pretrained was deprecated and redirected to speechbrain.inference in SpeechBrain 1.0, but required torchaudio >= 2.1.0
This commit is contained in:
@ -19,10 +19,11 @@ COPY requirements.txt .
|
||||
RUN pip install --no-cache-dir -r requirements.txt
|
||||
|
||||
# Install CUDA-optimized PyTorch (overwrites CPU versions)
|
||||
# Updated to torch 2.1.0+ for SpeechBrain 1.0 / pyannote diarization compatibility
|
||||
RUN pip install --force-reinstall \
|
||||
torch==2.0.1+cu118 \
|
||||
torchvision==0.15.2+cu118 \
|
||||
torchaudio==2.0.2+cu118 \
|
||||
torch==2.1.0+cu118 \
|
||||
torchvision==0.16.0+cu118 \
|
||||
torchaudio==2.1.0+cu118 \
|
||||
--index-url https://download.pytorch.org/whl/cu118
|
||||
|
||||
# Copy application code
|
||||
|
||||
Reference in New Issue
Block a user