forked from sambanova/ai-starter-kit
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.yaml
30 lines (25 loc) · 956 Bytes
/
config.yaml
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
api: "sncloud" # set either sncloud or sambastudio
embedding_model:
"type": "cpu" # set either sambastudio or cpu
"batch_size": 1 #set depending of your endpoint configuration (1 if CoE embedding expert)
"coe": True #set true if using Sambastudio embeddings in a CoE endpoint
"select_expert": "e5-mistral-7b-instruct" #set if using SambaStudio CoE embedding expert
llm:
"temperature": 0.1
"max_tokens_to_generate": 1000
"do_sample": False
"coe": True #set as true if using Sambastudio CoE endpoint
"select_expert": "llama3-405b" #set if using SambaStudio CoE llm expert
#sncloud CoE expert name -> "llama3-405b"
retrieval:
"chunk_size": 500
"chunk_overlap": 50
"db_type": "chroma"
"n_retrieved_documents": 3
query_decomposition:
"n_generated_subquestions": 3
sec:
"email": "mlengineer@snova_dummy.ai"
"company": "snova_dummy"
"report_type": "10-K"
"last_n_documents": 1