OAI: Remove unused parameters

Seed and low_mem aren't used, so comment them out.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri
2023-12-15 14:53:12 -05:00
parent 59729e2a4a
commit ed868fd262
2 changed files with 2 additions and 2 deletions

View File

@@ -28,7 +28,7 @@ class CommonCompletionRequest(BaseModel):
user: Optional[str] = None
# Generation info
seed: Optional[int] = -1
# seed: Optional[int] = -1
stream: Optional[bool] = False
stop: Optional[Union[str, List[str]]] = []

View File

@@ -36,7 +36,7 @@ class ModelLoadRequest(BaseModel):
rope_scale: Optional[float] = 1.0
rope_alpha: Optional[float] = 1.0
no_flash_attention: Optional[bool] = False
low_mem: Optional[bool] = False
# low_mem: Optional[bool] = False
prompt_template: Optional[str] = None
draft: Optional[DraftModelLoadRequest] = None