yuhaofeng-shiba commited on
Commit
06db57c
·
1 Parent(s): c633c41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -17,8 +17,10 @@ def init_args():
17
  global args
18
  parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
19
  args = parser.parse_args()
20
- args.load_model_path = './model_file/chatflow_13b.bin'
21
- args.config_path = './config/llama_13b_config.json'
 
 
22
  args.spm_model_path = './model_file/tokenizer.model'
23
  args.batch_size = 1
24
  args.seq_length = 1024
 
17
  global args
18
  parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
19
  args = parser.parse_args()
20
+ args.load_model_path = './model_file/chatllama_7b.bin'
21
+ args.config_path = './config/llama_7b.json'
22
+ # args.load_model_path = './model_file/chatflow_13b.bin'
23
+ # args.config_path = './config/llama_13b_config.json'
24
  args.spm_model_path = './model_file/tokenizer.model'
25
  args.batch_size = 1
26
  args.seq_length = 1024