diff options
Diffstat (limited to 'rag/generator/__init__.py')
| -rw-r--r-- | rag/generator/__init__.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/rag/generator/__init__.py b/rag/generator/__init__.py index 541eff8..ba23ffc 100644 --- a/rag/generator/__init__.py +++ b/rag/generator/__init__.py @@ -1,8 +1,8 @@ from typing import Type from .abstract import AbstractGenerator -from .ollama import Ollama from .cohere import Cohere +from .ollama import Ollama MODELS = ["ollama", "cohere"] |