Imsachinsingh00 commited on
Commit
9f1a73b
·
1 Parent(s): d4ac767

check the torch issue

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -1
Dockerfile CHANGED
@@ -31,7 +31,10 @@ RUN pip install --no-cache-dir -r requirements.txt
31
 
32
  # After installing requirements
33
  RUN pip uninstall -y torch torchvision && \
34
- pip install --no-cache-dir torch==2.6.0 torchvision==0.21.0
 
 
 
35
 
36
 
37
  # Explicitly install OpenAI Whisper
 
31
 
32
  # After installing requirements
33
  RUN pip uninstall -y torch torchvision && \
34
+ pip install --upgrade --force-reinstall --no-cache-dir torch==2.3.0 torchvision==0.18.0
35
+
36
+
37
+ RUN pip freeze | grep -E 'torch|torchvision'
38
 
39
 
40
  # Explicitly install OpenAI Whisper