This commit is contained in:
2024-07-18 15:22:19 +08:00
parent b7584b2a04
commit 5f18f0d3c1

View File

@@ -79,7 +79,12 @@ EOT
RUN <<EOT
#!/bin/bash
pip install -v -U git+https://github.com/facebookresearch/xformers.git@main#egg=xformers
EOT
RUN <<EOT
#!/bin/bash
pip install git+https://github.com/huggingface/transformers
python -c "from transformers import pipeline; print(pipeline('sentiment-analysis')('I love you'))"
EOT
RUN <<EOT