Merge pull request #157 from VinciGit00/llava_integration

VinciGit00 · web-flow · commit f6442ccca357 · 2024-05-06T14:34:32.000+02:00
diff --git a/.gitignore b/.gitignore
@@ -36,3 +36,4 @@ poetry.lock
 # lock files
 *.lock
 poetry.lock
+ 
diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py
@@ -32,6 +32,8 @@
     "ollama": {
         "llama2": 4096,
         "llama3": 8192,
+        "llava": 4096,
+        "llava_next": 4096,
         "mistral": 8192,
         "codellama": 16000,
         "dolphin-mixtral": 32000,
