diff --git a/pilot/__init__.py b/pilot/__init__.py index b6865c44c..b1d1cd3d2 100644 --- a/pilot/__init__.py +++ b/pilot/__init__.py @@ -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" diff --git a/pilot/configs/model_config.py b/pilot/configs/model_config.py index af6c138b5..ad81a6e79 100644 --- a/pilot/configs/model_config.py +++ b/pilot/configs/model_config.py @@ -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") \ No newline at end of file diff --git a/pilot/server/webserver.py b/pilot/server/webserver.py index 3ed989b07..233655381 100644 --- a/pilot/server/webserver.py +++ b/pilot/server/webserver.py @@ -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)