Pin transformers to a working Llama-compatible version

This commit is contained in:
0cc4m
2023-03-24 19:07:28 +00:00
parent 026eb3205e
commit 9dcba38978
2 changed files with 2 additions and 2 deletions

View File

@@ -30,7 +30,7 @@ dependencies:
- flask-ngrok - flask-ngrok
- flask-cors - flask-cors
- lupa==1.10 - lupa==1.10
- git+https://github.com/zphang/transformers@llama_push - git+https://github.com/huggingface/transformers@88dae78f4d204428568f749e864ef5ba09da7d24
- git+https://github.com/huggingface/datasets@test-pandas-2.0.0rc - git+https://github.com/huggingface/datasets@test-pandas-2.0.0rc
- huggingface_hub==0.12.1 - huggingface_hub==0.12.1
- safetensors - safetensors

View File

@@ -29,7 +29,7 @@ dependencies:
- flask-ngrok - flask-ngrok
- flask-cors - flask-cors
- lupa==1.10 - lupa==1.10
- transformers==4.25.1 - git+https://github.com/huggingface/transformers@88dae78f4d204428568f749e864ef5ba09da7d24
- huggingface_hub==0.12.1 - huggingface_hub==0.12.1
- safetensors - safetensors
- accelerate - accelerate