SoSa123456 commited on
Commit
e2f86d9
1 Parent(s): e350155

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -10,7 +10,7 @@ RUN sudo apt-get update -y
10
  RUN sudo apt-get upgrade -y
11
  # Install any needed packages specified in requirements.txt
12
  RUN python -m pip install --upgrade pip
13
- RUN pip install transformers[torch]
14
  RUN pip install -r requirements.txt
15
 
16
  RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
 
10
  RUN sudo apt-get upgrade -y
11
  # Install any needed packages specified in requirements.txt
12
  RUN python -m pip install --upgrade pip
13
+ RUN pip install transformers[torch] -y
14
  RUN pip install -r requirements.txt
15
 
16
  RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118