feat: 模型下载目录修改

jiaoqiyuan · Dec 18, 2023 · 51a31b5 · 51a31b5
1 parent 4a86dad
commit 51a31b5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/install_model.py b/install_model.py
@@ -12,8 +12,7 @@
 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
@@ -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;