zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 模型下载目录修改

Browse Source
This commit is contained in:
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': {

4
main.py
View File

@ -44,11 +44,11 @@ def perform_db_migrate():
def start_services(): def start_services():
management.call_command('migrate') management.call_command('migrate')
management.call_command('runserver',"0.0.0.0:8000") management.call_command('runserver', "0.0.0.0:8000")
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;