diff --git a/src/models/embedding.py b/src/models/embedding.py
index 7b576abe..604d1c1d 100644
--- a/src/models/embedding.py
+++ b/src/models/embedding.py
@@ -15,11 +15,7 @@ GLOBAL_EMBED_STATE = {}
class EmbeddingModel(FlagModel):
def __init__(self, model_info, config, **kwargs):
self.info = model_info
- model_name_or_path = handle_local_model(
- paths=config.model_local_paths,
- model_name=model_info["name"],
- default_path=model_info.get("default_path", None))
-
+ model_name_or_path = config.model_local_paths.get(model_info["name"], model_info.get("default_path", None))
logger.info(f"Loading embedding model {model_info['name']} from {model_name_or_path}")
super().__init__(model_name_or_path,
@@ -34,11 +30,7 @@ class Reranker(FlagReranker):
assert config.reranker in RERANKER_LIST.keys(), f"Unsupported Reranker: {config.reranker}, only support {RERANKER_LIST.keys()}"
- model_name_or_path = handle_local_model(
- paths=config.model_local_paths,
- model_name=config.reranker,
- default_path=RERANKER_LIST[config.reranker])
-
+ model_name_or_path = config.model_local_paths.get(config.reranker, default_path=RERANKER_LIST[config.reranker])
logger.info(f"Loading Reranker model {config.reranker} from {model_name_or_path}")
super().__init__(model_name_or_path, use_fp16=True, **kwargs)
@@ -113,6 +105,4 @@ def get_embedding_model(config):
def handle_local_model(paths, model_name, default_path):
model_path = paths.get(model_name, default_path)
- if os.getenv("MODEL_ROOT_DIR") and not os.path.isabs(model_path):
- model_path = os.path.join(os.getenv("MODEL_ROOT_DIR"), model_path)
return model_path
\ No newline at end of file
diff --git a/src/plugins/oneke.py b/src/plugins/oneke.py
index aea4c5c8..3ca3d60b 100644
--- a/src/plugins/oneke.py
+++ b/src/plugins/oneke.py
@@ -16,8 +16,6 @@ logger = setup_logger("OneKE")
dotenv.load_dotenv()
-MODEL_NAME_OR_PATH = os.path.join(os.getenv('MODEL_ROOT_DIR', './'), 'OneKE')
-
instruction_mapper = {
'NERzh': "你是专门进行实体抽取的专家。请从input中抽取出符合schema定义的实体,不存在的实体类型返回空列表。请按照JSON字符串的格式回答。",
'REzh': "你是专门进行关系抽取的专家。请从input中抽取出符合schema定义的关系三元组。请按照JSON字符串的格式回答。",
@@ -42,7 +40,7 @@ class OneKE:
def __init__(self, config=None):
self.config = config
- model_name_or_path = config.model_local_paths.get('oneke', "zjunlp/OneKE")
+ model_name_or_path = config.model_local_paths.get('zjunlp/OneKE', "zjunlp/OneKE")
logger.info(f"Loading KGC model OneKE from {model_name_or_path}")
model_config = AutoConfig.from_pretrained(model_name_or_path, trust_remote_code=True)
diff --git a/web/src/components/TableConfigComponent.vue b/web/src/components/TableConfigComponent.vue
new file mode 100644
index 00000000..7b372d4c
--- /dev/null
+++ b/web/src/components/TableConfigComponent.vue
@@ -0,0 +1,202 @@
+
+