github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-30 23:28:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

cfg: update

Browse Source
This commit is contained in:
csunny 2023-05-18 17:32:53 +08:00
parent 9ddd089381
commit 0d370a1a64
3 changed files with 2 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pilot/__init__.py
View File

@ -1,11 +1,5 @@
from pilot.source_embedding import (SourceEmbedding, register)
import os
import sys
ROOT_PATH = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
sys.path.append(ROOT_PATH)
__all__ = [
"SourceEmbedding",
"register"

7
pilot/configs/model_config.py
View File

@ -28,12 +28,5 @@ ISDEBUG = False
VECTOR_SEARCH_TOP_K = 3
# LLM_MODEL = "vicuna-13b"
# LIMIT_MODEL_CONCURRENCY = 5
# MAX_POSITION_EMBEDDINGS = 4096
# VICUNA_MODEL_SERVER = "http://121.41.167.183:8000"
VS_ROOT_PATH = os.path.join(os.path.dirname(os.path.dirname(__file__)), "vs_store")
KNOWLEDGE_UPLOAD_ROOT_PATH = os.path.join(os.path.dirname(os.path.dirname(__file__)), "knowledge")

4
pilot/server/webserver.py
View File

@ -6,6 +6,7 @@ import os
import shutil
import uuid
import json
import sys
import time
import gradio as gr
import datetime
@ -13,8 +14,7 @@ import requests
from urllib.parse import urljoin
from langchain import PromptTemplate
import os
import sys
ROOT_PATH = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
sys.path.append(ROOT_PATH)