Skip to content

Commit

Permalink
include default comments
Browse files Browse the repository at this point in the history
  • Loading branch information
NickyHavoc committed Apr 24, 2024
1 parent e0a64fa commit b860b19
Showing 1 changed file with 7 additions and 4 deletions.
11 changes: 7 additions & 4 deletions src/intelligence_layer/core/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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__(
Expand Down Expand Up @@ -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(
Expand Down Expand Up @@ -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("""<s>[INST] <<SYS>>
Expand Down Expand Up @@ -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(
Expand Down

0 comments on commit b860b19

Please sign in to comment.