CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-07-01 21:00:48 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(models_tokens): add llama2 and llama3 sizes explicitly

Browse Source
This commit is contained in:
Federico Aguzzi 2024-08-21 09:32:59 +02:00
parent dd406dff0c
commit b05ec16b25
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scrapegraphai/helpers/models_tokens.py
View File

@ -58,7 +58,11 @@ models_tokens = {
"deepseek-coder:33b": 16000, "deepseek-coder:33b": 16000,
"falcon": 2048, "falcon": 2048,
"llama2": 4096, "llama2": 4096,
"llama2:7b": 4096,
"llama2:13b": 4096,
"llama2:70b": 4096,
"llama3": 8192, "llama3": 8192,
"llama3:8b": 8192,
"llama3:70b": 8192, "llama3:70b": 8192,
"llama3.1":128000, "llama3.1":128000,
"llama3.1:8b": 128000, "llama3.1:8b": 128000,