From 51a31b5e824ed7fd605b293b67180132681c2b67 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Mon, 18 Dec 2023 16:13:37 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=A8=A1=E5=9E=8B=E4=B8=8B=E8=BD=BD?= =?UTF-8?q?=E7=9B=AE=E5=BD=95=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- install_model.py | 3 +-- main.py | 4 ++-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/install_model.py b/install_model.py index 1400cf93f..d9395bc5d 100644 --- a/install_model.py +++ b/install_model.py @@ -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': { diff --git a/main.py b/main.py index 8f8802302..1a702706f 100644 --- a/main.py +++ b/main.py @@ -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;