File size: 397 Bytes
f1a8641
 
 
6176da6
9331285
f1a8641
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
MODEL_CONFIG = {
#    "model_name": "deepseek-ai/DeepSeek-R1-Distill-Qwen-7B", #"Qwen/Qwen2.5-1.5B-Instruct",
    # "model_name": "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B",
    "model_name": "scott12355/Qwen-SophiaAI-Finetune-7B",
    "max_new_tokens": 350,
    "num_return_sequences": 1,
    "batch_size": 8,
    "max_conversation_history_size": 100
} 

RAG_CONFIG = {
    "path": "documents"
}