lucianotonet commited on
Commit
71321a9
1 Parent(s): f0e36a6

Atualiza dependências do Docker para incluir Transformers

Browse files

Adiciona a instalação do pacote Transformers diretamente do repositório do GitHub, removendo a versão específica do arquivo de requisitos. Essa mudança garante que a versão mais recente seja utilizada, melhorando a compatibilidade e aproveitando novas funcionalidades e correções de bugs. Além disso, facilita a atualização do ambiente, alinhando-o com as últimas inovações da biblioteca.

Files changed (2) hide show
  1. Dockerfile +2 -1
  2. requirements.txt +0 -1
Dockerfile CHANGED
@@ -5,7 +5,8 @@ FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
5
 
6
  # Instalar dependências
7
  COPY requirements.txt ./
8
- RUN pip install --no-cache-dir -r requirements.txt
 
9
 
10
  # Definir variáveis de ambiente
11
  ENV HUGGINGFACE_HUB_CACHE=/home/user/.cache/huggingface
 
5
 
6
  # Instalar dependências
7
  COPY requirements.txt ./
8
+ RUN pip install --no-cache-dir -r requirements.txt && \
9
+ pip install --no-cache-dir git+https://github.com/huggingface/transformers
10
 
11
  # Definir variáveis de ambiente
12
  ENV HUGGINGFACE_HUB_CACHE=/home/user/.cache/huggingface
requirements.txt CHANGED
@@ -1,4 +1,3 @@
1
- transformers>=4.37.0
2
  accelerate
3
  safetensors
4
  sentencepiece
 
 
1
  accelerate
2
  safetensors
3
  sentencepiece