From a95aabe1ac489ffbc291e439e2bb92890c48f1b4 Mon Sep 17 00:00:00 2001 From: Jonathan Dunn Date: Tue, 12 Mar 2024 12:43:41 -0400 Subject: [PATCH] fixed an error with -ChangeDefaultModel with local models --- installer/client/cli/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/installer/client/cli/utils.py b/installer/client/cli/utils.py index 9757ac7..b75cd64 100644 --- a/installer/client/cli/utils.py +++ b/installer/client/cli/utils.py @@ -447,7 +447,7 @@ class Setup: try: default_modelollamaList = ollama.list()['models'] for model in default_modelollamaList: - self.fullOllamaList.append(model['name'].rstrip(":latest")) + self.fullOllamaList.append(model['name']) except: self.fullOllamaList = [] allmodels = self.gptlist + self.fullOllamaList + self.claudeList