Bump transformers to latest version so it can load the llava_mistral models

2023-10-20 10:04:27 +02:00
parent cbb295ea1f
commit bfc8345434
1 changed files with 1 additions and 0 deletions
--- a/1
+++ b/1
@@ -77,6 +77,7 @@ RUN source /venv/bin/activate && \
    pip3 install -e . && \
    pip3 install ninja && \
    pip3 install flash-attn --no-build-isolation && \
+    pip3 install transformers==4.34.1 && \
    deactivate

 # Install Jupyter