invenxion-chatbot_local

Sleeping

Ibraaheem commited on Jan 6, 2024

Commit

c5b591f

1 Parent(s): 42247c8

Update private_gpt/settings/settings.py

Files changed (1) hide show

private_gpt/settings/settings.py CHANGED Viewed

@@ -81,7 +81,7 @@ class DataSettings(BaseModel):
 class LLMSettings(BaseModel):
-    mode: Literal["local", "openai", "sagemaker", "mock"]
     max_new_tokens: int = Field(
         256,
         description="The maximum number of token that the LLM is authorized to generate in one completion.",

 class LLMSettings(BaseModel):
+    mode: Literal["local", "openai", "sagemaker", "mock", "dynamic"]
     max_new_tokens: int = Field(
         256,
         description="The maximum number of token that the LLM is authorized to generate in one completion.",