[1bdb11]: / aitrika / config / config.py

Download this file

20 lines (14 with data), 319 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
"""
Default configs.
"""
from dataclasses import dataclass
@dataclass(frozen=True)
class Config:
ENTREZ_EMAIL = "mail@mail.com"
@dataclass(frozen=True)
class LLMConfig:
DEFAULT_EMBEDDINGS = "allenai/specter2_base"
CHUNK_SIZE = 1024
CHUNK_OVERLAP = 80
CONTEXT_WINDOW = 2048
NUM_OUTPUT = 256