fixed something with llama models

danielmiessler · Mar 11, 2024 · 220bb4e · 220bb4e
1 parent 9b26ca6
commit 220bb4e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/installer/client/cli/utils.py b/installer/client/cli/utils.py
@@ -278,7 +278,7 @@ def fetch_available_models(self):
         try:
             default_modelollamaList = ollama.list()['models']
             for model in default_modelollamaList:
-                fullOllamaList.append(model['name'].rstrip(":latest"))
+                fullOllamaList.append(model['name'].split(":")[0])
         except:
             fullOllamaList = []
         return gptlist, fullOllamaList, claudeList