Updated docker image and requirements

Dockerfile

@@ -1,4 +1,8 @@
-FROM huggingface/transformers-pytorch-gpu AS requirements
+FROM nvidia/cuda:12.2.0-runtime-ubuntu22.04 AS requirements
+RUN apt-get update
+RUN apt-get update -y
+RUN apt install -y python3-pip
+ENV TOKENIZERS_PARALLELISM=false
 COPY requirements.txt /requirements.txt
 RUN pip3 install -r /requirements.txt
 RUN rm -rf /requirements.txt

requirements.txt

@@ -1,3 +1,4 @@
+accelerate
 bs4
 torch
 transformers