dillon_stuff/TalkEdit
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases 1 Wiki Activity

fix: Update PyTorch/torchaudio to 2.1.0 for SpeechBrain 1.0 diarization compatibility

Browse Source 
- 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:
Your Name
2026-01-05 09:57:20 -05:00
parent 81f40726e3
commit c5cc7c2969
3 changed files with 16 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -20,7 +20,7 @@ RUN pip install --no-cache-dir -r requirements.txt
# Optional: Install CUDA-specific PyTorch if GPU support needed
# Uncomment and modify for your CUDA version:
# RUN pip install --force-reinstall torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
# RUN pip install --force-reinstall 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
COPY . .