添加对复旦大学MOSS的支持

这个提交包含在:
binary-husky
2023-05-07 01:04:59 +08:00
父节点 4b9078a9dc
当前提交 5102ec8263
共有 4 个文件被更改,包括 269 次插入2 次删除

查看文件

@@ -173,7 +173,19 @@ if "jittorllms_pangualpha" in AVAIL_LLM_MODELS:
"token_cnt": get_token_num_gpt35,
},
})
if "moss" in AVAIL_LLM_MODELS:
from .bridge_moss import predict_no_ui_long_connection as moss_noui
from .bridge_moss import predict as moss_ui
model_info.update({
"moss": {
"fn_with_ui": moss_ui,
"fn_without_ui": moss_noui,
"endpoint": None,
"max_token": 1024,
"tokenizer": tokenizer_gpt35,
"token_cnt": get_token_num_gpt35,
},
})