From b860b196b8d574fb6103419ae6a9c893c945e4d5 Mon Sep 17 00:00:00 2001 From: "niklas.finken" Date: Wed, 24 Apr 2024 12:47:15 +0200 Subject: [PATCH] include default comments --- src/intelligence_layer/core/model.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/src/intelligence_layer/core/model.py b/src/intelligence_layer/core/model.py index 5e6980d3d..d64678dd7 100644 --- a/src/intelligence_layer/core/model.py +++ b/src/intelligence_layer/core/model.py @@ -140,7 +140,7 @@ class AlephAlphaModel: name: The name of a valid model that can access an API using an implementation of the AlephAlphaClientProtocol. client: Aleph Alpha client instance for running model related API calls. - Defaults to the :class:`LimitedConcurrencyClient` + Defaults to :class:`LimitedConcurrencyClient` """ def __init__( @@ -190,8 +190,9 @@ class LuminousControlModel(ControlModel): Args: name: The name of a valid model second generation control model. + Defaults to `luminous-base-control` client: Aleph Alpha client instance for running model related API calls. - Defaults to the :class:`LimitedConcurrencyClient` + Defaults to :class:`LimitedConcurrencyClient` """ INSTRUCTION_PROMPT_TEMPLATE = PromptTemplate( @@ -237,8 +238,9 @@ class Llama2InstructModel(ControlModel): Args: name: The name of a valid llama-2 model. + Defaults to `llama-2-13b-chat` client: Aleph Alpha client instance for running model related API calls. - Defaults to the :class:`LimitedConcurrencyClient` + Defaults to :class:`LimitedConcurrencyClient` """ INSTRUCTION_PROMPT_TEMPLATE = PromptTemplate("""[INST] <> @@ -276,8 +278,9 @@ class Llama3InstructModel(ControlModel): Args: name: The name of a valid llama-3 model. + Defaults to `llama-3-8b-instruct` client: Aleph Alpha client instance for running model related API calls. - Defaults to the :class:`LimitedConcurrencyClient` + Defaults to :class:`LimitedConcurrencyClient` """ INSTRUCTION_PROMPT_TEMPLATE = PromptTemplate(