user_config.py

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.

# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.

# isort: skip_file
from models.dummy_model import DummyModel
GenerateModel = DummyModel
# EvaluateModel = DummyModel

# Uncomment the lines below to use the Vanilla LLAMA baseline
# from models.vanilla_llama_baseline import InstructModel
# GenerateModel = InstructModel
# EvaluateModel = InstructModel



# Uncomment the lines below to use the RAG LLAMA baseline
# from models.rag_llama_baseline import RAGModel
# GenerateModel = RAGModel
# EvaluateModel = RAGModel

# Uncomment the lines below to use the RAG KG LLAMA baseline
# from models.rag_knowledge_graph_baseline import RAG_KG_Model
# GenerateModel = RAG_KG_Model
# EvaluateModel = RAG_KG_Model


# from models.openapi_model import OpenAIModel
# EvaluateModel = OpenAIModel

from models.ollama_model import OllamaModel
EvaluateModel = OllamaModel