From 95305f59df84caded50286b1a57b6075e48725a8 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Wed, 24 Apr 2024 01:10:43 +0200 Subject: Rerank working llama3 sucks at rag --- rag/generator/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'rag/generator/__init__.py') 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() -- cgit v1.2.3-70-g09d2