diff --git a/Dockerfile b/Dockerfile index 366e500..c6d22b5 100644 --- a/Dockerfile +++ b/Dockerfile @@ -16,7 +16,8 @@ RUN apt-get update && apt-get install -y git \ CUDA_HOME=/usr/local/cuda \ TORCH_CUDA_ARCH_LIST="10.0" \ pip install --no-cache-dir --no-build-isolation . && \ - rm -rf /tmp/lmcache + rm -rf /tmp/lmcache && \ + pip install --no-cache-dir pandas # Copy over nemotron reasonong parser COPY ./super_v3_reasoning_parser.py /opt/super_v3_reasoning_parser.py