diff --git a/Dockerfile b/Dockerfile
index 17d0b00..366f19c 100644
--- a/Dockerfile
+++ b/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
diff --git a/requirements.txt b/requirements.txt
index ed79fe8..309cbd5 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
+accelerate
 bs4
 torch
 transformers
\ No newline at end of file