From 1d3a76c2cf099fbf573a4dcbd5312bd06f470a60 Mon Sep 17 00:00:00 2001
From: pandyamarut <pandyamarut@gmail.com>
Date: Wed, 18 Sep 2024 17:18:34 -0700
Subject: [PATCH] update dockerfile

Signed-off-by: pandyamarut <pandyamarut@gmail.com>
---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 1423c29..506397e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -20,7 +20,7 @@ WORKDIR /app/TensorRT-LLM/examples/llm-api
 RUN pip3 install -r requirements.txt
 
 # Install additional dependencies for the serverless worker
-RUN pip3 install runpod
+RUN pip3 install --upgrade runpod transformers
 
 # Set the working directory to /app
 WORKDIR /app