diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-24 01:10:43 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-24 01:10:43 +0200 |
commit | 95305f59df84caded50286b1a57b6075e48725a8 (patch) | |
tree | c0f0157a99da6332a3c96462b0aba2bd02dfcb33 /rag/generator/__init__.py | |
parent | 75be0914f6bd2cdeda1539f83b38fcbc854d5cfa (diff) |
Rerank working
llama3 sucks at rag
Diffstat (limited to 'rag/generator/__init__.py')
-rw-r--r-- | rag/generator/__init__.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/rag/generator/__init__.py b/rag/generator/__init__.py index ba23ffc..a776231 100644 --- a/rag/generator/__init__.py +++ b/rag/generator/__init__.py @@ -4,11 +4,11 @@ from .abstract import AbstractGenerator from .cohere import Cohere from .ollama import Ollama -MODELS = ["ollama", "cohere"] +MODELS = ["local", "cohere"] def get_generator(model: str) -> Type[AbstractGenerator]: match model: - case "ollama": + case "local": return Ollama() case "cohere": return Cohere() |