diff --git a/installer/client/cli/utils.py b/installer/client/cli/utils.py index 94dea99b7..5bc10c22d 100644 --- 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