Skip to content

Commit d5547a4

Browse files
authored
feat: add new hugging_face models
1 parent 8c7c3e3 commit d5547a4

File tree

1 file changed

+30
-1
lines changed

1 file changed

+30
-1
lines changed

scrapegraphai/helpers/models_tokens.py

Lines changed: 30 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,5 +71,34 @@
7171
"mistral": {
7272
"mistralai/Mistral-7B-Instruct-v0.2": 32000
7373
},
74-
"hugging_face": {}
74+
"hugging_face": {
75+
"meta-llama/Meta-Llama-3-8B": 8192,
76+
"meta-llama/Meta-Llama-3-8B-Instruct": 8192,
77+
"meta-llama/Meta-Llama-3-70B": 8192,
78+
"meta-llama/Meta-Llama-3-70B-Instruct": 8192,
79+
"google/gemma-2b": 8192,
80+
"google/gemma-2b-it": 8192,
81+
"google/gemma-7b": 8192,
82+
"google/gemma-7b-it": 8192,
83+
"microsoft/phi-2": 2048,
84+
"openai-community/gpt2": 1024,
85+
"openai-community/gpt2-medium": 1024,
86+
"openai-community/gpt2-large": 1024,
87+
"facebook/opt-125m": 2048,
88+
"petals-team/StableBeluga2": 8192,
89+
"distilbert/distilgpt2": 1024,
90+
"mistralai/Mistral-7B-Instruct-v0.2": 32768,
91+
"gradientai/Llama-3-8B-Instruct-Gradient-1048k": 1040200,
92+
"NousResearch/Hermes-2-Pro-Llama-3-8B": 8192,
93+
"NousResearch/Hermes-2-Pro-Llama-3-8B-GGUF": 8192,
94+
"nvidia/Llama3-ChatQA-1.5-8B": 8192,
95+
"microsoft/Phi-3-mini-4k-instruct": 4192,
96+
"microsoft/Phi-3-mini-128k-instruct": 131072,
97+
"mlabonne/Meta-Llama-3-120B-Instruct": 8192,
98+
"cognitivecomputations/dolphin-2.9-llama3-8b": 8192,
99+
"cognitivecomputations/dolphin-2.9-llama3-8b-gguf": 8192,
100+
"cognitivecomputations/dolphin-2.8-mistral-7b-v02": 32768,
101+
"cognitivecomputations/dolphin-2.5-mixtral-8x7b": 32768,
102+
"TheBloke/dolphin-2.7-mixtral-8x7b-GGUF": 32768
103+
}
75104
}

0 commit comments

Comments
 (0)