Add low_vram to server settings

This commit is contained in:
Andrei Betlen 2023-06-14 22:13:42 -04:00
parent 44b83cada5
commit 1e20be6d0c

View file

@ -48,6 +48,10 @@ class Settings(BaseSettings):
description="Use mmap.", description="Use mmap.",
) )
embedding: bool = Field(default=True, description="Whether to use embeddings.") embedding: bool = Field(default=True, description="Whether to use embeddings.")
low_vram: bool = Field(
default=False,
description="Whether to use less VRAM. This will reduce performance.",
)
last_n_tokens_size: int = Field( last_n_tokens_size: int = Field(
default=64, default=64,
ge=0, ge=0,