-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.yml
73 lines (72 loc) · 2.01 KB
/
config.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
api:
host: '127.0.0.1'
port: 5000
llm:
model:
type: "llama"
repo: "TheBloke/Nous-Hermes-Llama2-GGUF"
filename: "nous-hermes-llama2-13b.Q6_K.gguf"
max_response_tokens: 1024
n_batch: 2048
n_ctx: 4096
n_gpu_layers: -1
repeat_penalty: 1.176
temperature: 0.71
top_k: 40
top_p: 0.1
verbose: True
data_dir: '/home/core/llm/chatbot/data'
rag:
libpages:
name: 'libpages'
rag:
name: 'libpages'
stack:
module_name: 'deckard.rag'
class_name: 'RagStack'
collectors:
-
name: 'Library Pages'
module_name: 'deckard.collectors'
class_name: 'LibPagesCollector'
output: 'collector/libpages/data/output'
config:
cache_urls: True
ignore_links: True
images_to_alt: True
unicode_snob: True
context:
size: 4096
max_vector_distance: 1.0
chunker:
module_name: 'deckard.chunkers'
class_name: 'CharacterTextSplitterChunker'
config:
chunk_size: 1024
overlap: 128
split_on: "\n"
add_document_metadata: False
write_chunks_to_disk: True
context_builder:
module_name: 'deckard.context_builders'
class_name: 'SimpleContextAggregator'
context_database:
module_name: 'deckard.databases'
class_name: 'SQLite'
name: 'libpages'
embedding_database:
module_name: 'deckard.databases'
class_name: 'LanceDB'
name: 'libpages'
embedding_encoder:
module_name: 'deckard.encoders'
class_name: 'SentenceTransformerEncoder'
model: 'avsolatorio/GIST-large-Embedding-v0'
query_processor:
module_name: 'deckard.query_processors'
class_name: 'StandardQueryProcessor'
reranker:
module_name: 'deckard.encoders'
class_name: 'SentenceTransformerEncoder'
model: 'BAAI/bge-reranker-large'
max_raw_results: 10