vector2000 commited on
Commit
1c31b34
·
verified ·
1 Parent(s): 28c15ef

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -12,6 +12,7 @@ RUN pip install --upgrade pip
12
 
13
  # Встановлення необхідних пакетів
14
  RUN pip install torch transformers datasets tqdm
 
15
 
16
  # Скопіювати ваш Python-код у контейнер
17
  COPY . /app
 
12
 
13
  # Встановлення необхідних пакетів
14
  RUN pip install torch transformers datasets tqdm
15
+ RUN pip install transformers[torch]
16
 
17
  # Скопіювати ваш Python-код у контейнер
18
  COPY . /app