diff --git a/deepeval/models/hallucination_model.py b/deepeval/models/hallucination_model.py index 5e3f48464..04862a4f9 100644 --- a/deepeval/models/hallucination_model.py +++ b/deepeval/models/hallucination_model.py @@ -2,11 +2,9 @@ from deepeval.singleton import Singleton from sentence_transformers import CrossEncoder from deepeval.progress_context import progress_context -from deepeval.models.model_map import model_map, name_to_card - class HallucinationModel(metaclass=Singleton): - def __init__(self, model_name: str = "vectara-hallucination"): + def __init__(self, model_name: str = "vectara/hallucination_evaluation_model"): # We use a smple cross encoder model os.environ["TOKENIZERS_PARALLELISM"] = "false" @@ -14,5 +12,4 @@ def __init__(self, model_name: str = "vectara-hallucination"): with progress_context( "Downloading HallucinationEvaluationModel (may take up to 2 minutes if running for the first time)..." ): - model_name = name_to_card(model_name) self.model = CrossEncoder(model_name)