This commit is contained in:
Chen, Zhentao 2024-02-23 16:33:36 +08:00
parent 5399343adc
commit e1fcf54a0c

View file

@ -161,9 +161,9 @@ jobs:
run: |
pip install --upgrade datasets==2.14.6
if [ "${{ matrix.model_name }}" = "Mistral-7B-v0.1" ]; then
pip install --upgrade transformers==4.36
pip install --upgrade transformers==4.36
else
pip install --upgrade transformers==4.31
pip install --upgrade transformers==4.31
fi