15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-25 17:22:55 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

feat: 模型下载目录修改

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
import sentence_transformers
prefix_dir = os.path.join(os.path.dirname(os.path.abspath(os.getcwd())), 'model')
prefix_dir = "/opt/maxkb/model"
model_config = [
{
'download_params': {

4
main.py
View File

@ -44,11 +44,11 @@ def perform_db_migrate():
def start_services():
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__':
os.environ['TRANSFORMERS_CACHE'] = '/opt/maxkb/model'
os.environ['TRANSFORMERS_CACHE'] = '/opt/maxkb/model/base'
parser = argparse.ArgumentParser(
description="""
qabot service control tools;