kadirnar commited on
Commit
c77b6a0
1 Parent(s): c6cf7f1

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -1
Dockerfile CHANGED
@@ -36,7 +36,9 @@ RUN pyenv install $PYTHON_VERSION && \
36
  pyenv global $PYTHON_VERSION && \
37
  pyenv rehash && \
38
  pip install --no-cache-dir --upgrade pip setuptools wheel && \
39
- pip install --no-cache-dir
 
 
40
 
41
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
42
  # PyTorch installation with CUDA 12.1 support
 
36
  pyenv global $PYTHON_VERSION && \
37
  pyenv rehash && \
38
  pip install --no-cache-dir --upgrade pip setuptools wheel && \
39
+ pip install --no-cache-dir \
40
+ datasets \
41
+ huggingface-hub "protobuf<4" "click<8.1"
42
 
43
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
44
  # PyTorch installation with CUDA 12.1 support