panyy
/
UnisKB
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: 模型下载目录修改

v3.2
shaohuzhang1 2023-12-18 16:13:37 +08:00
parent 4a86dad733
commit 51a31b5e82
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
install_model.py
View File

@ -12,8 +12,7 @@ import os.path
from transformers import GPT2TokenizerFast from transformers import GPT2TokenizerFast
import sentence_transformers import sentence_transformers
prefix_dir = os.path.join(os.path.dirname(os.path.abspath(os.getcwd())), 'model') prefix_dir = "/opt/maxkb/model"
model_config = [ model_config = [
{ {
'download_params': { 'download_params': {

2
main.py
View File

@ -48,7 +48,7 @@ def start_services():
if __name__ == '__main__': if __name__ == '__main__':
os.environ['TRANSFORMERS_CACHE'] = '/opt/maxkb/model' os.environ['TRANSFORMERS_CACHE'] = '/opt/maxkb/model/base'
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
description=""" description="""
qabot service control tools; qabot service control tools;