feat: 模型下载目录修改

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': {

main.py

@@ -48,7 +48,7 @@ def start_services():
 
 
 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;