CDOM201 commited on
Commit
a64cb3d
·
verified ·
1 Parent(s): a82b46c

Matching req with versions

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -2
  2. requirements.txt +0 -4
Dockerfile CHANGED
@@ -1,5 +1,5 @@
1
- # Use an official PyTorch image with CUDA support
2
- FROM pytorch/pytorch:2.5.1-cuda11.8-cudnn9-runtime
3
 
4
  # Set environment variables
5
  ENV PYTHONUNBUFFERED=1 \
 
1
+ # Use a newer PyTorch image that matches chatterbox-tts requirements
2
+ FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-runtime
3
 
4
  # Set environment variables
5
  ENV PYTHONUNBUFFERED=1 \
requirements.txt CHANGED
@@ -1,7 +1,3 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu118
2
- torch==2.5.1+cu118
3
- torchaudio==2.5.1+cu118
4
- torchvision==0.20.1+cu118
5
  fastapi==0.127.0
6
  uvicorn==0.40.0
7
  pydantic==2.11.10
 
 
 
 
 
1
  fastapi==0.127.0
2
  uvicorn==0.40.0
3
  pydantic==2.11.10