Skip to content

Commit 41ea3e0

Browse files
committed
add llama3.1 model
1 parent 72ee93a commit 41ea3e0

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

examples/local_models/smart_scraper_ollama.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
graph_config = {
1111
"llm": {
12-
"model": "ollama/llama3",
12+
"model": "ollama/llama3.1",
1313
"temperature": 0,
1414
"format": "json", # Ollama needs the format to be specified explicitly
1515
# "base_url": "http://localhost:11434", # set ollama URL arbitrarily

scrapegraphai/helpers/models_tokens.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,10 @@
5151
"falcon": 2048,
5252
"llama2": 4096,
5353
"llama3": 8192,
54+
"llama3:70b": 8192,
55+
"llama3.1":128000,
56+
"llama3.1:70b": 128000,
57+
"lama3.1:405b": 128000,
5458
"scrapegraph": 8192,
5559
"llava": 4096,
5660
"mixtral:8x22b-instruct": 65536,

0 commit comments

Comments
 (0)