mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-10-22 09:28:42 +00:00
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding:utf-8 -*-
|
|
|
|
import torch
|
|
import os
|
|
import nltk
|
|
|
|
|
|
ROOT_PATH = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
MODEL_PATH = os.path.join(ROOT_PATH, "models")
|
|
PILOT_PATH = os.path.join(ROOT_PATH, "pilot")
|
|
VECTORE_PATH = os.path.join(PILOT_PATH, "vector_store")
|
|
LOGDIR = os.path.join(ROOT_PATH, "logs")
|
|
DATASETS_DIR = os.path.join(PILOT_PATH, "datasets")
|
|
DATA_DIR = os.path.join(PILOT_PATH, "data")
|
|
|
|
nltk.data.path = [os.path.join(PILOT_PATH, "nltk_data")] + nltk.data.path
|
|
|
|
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
|
LLM_MODEL_CONFIG = {
|
|
"flan-t5-base": os.path.join(MODEL_PATH, "flan-t5-base"),
|
|
"vicuna-13b": os.path.join(MODEL_PATH, "vicuna-13b"),
|
|
"sentence-transforms": os.path.join(MODEL_PATH, "all-MiniLM-L6-v2")
|
|
}
|
|
|
|
|
|
VECTOR_SEARCH_TOP_K = 3
|
|
LLM_MODEL = "vicuna-13b"
|
|
LIMIT_MODEL_CONCURRENCY = 5
|
|
MAX_POSITION_EMBEDDINGS = 4096
|
|
VICUNA_MODEL_SERVER = "http://47.97.125.199:8000"
|
|
|
|
# Load model config
|
|
ISLOAD_8BIT = True
|
|
ISDEBUG = False
|
|
|
|
|
|
DB_SETTINGS = {
|
|
"user": "root",
|
|
"password": "aa12345678",
|
|
"host": "localhost",
|
|
"port": 3306
|
|
} |