P01son commited on
Commit
6250012
·
1 Parent(s): 06db57c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -17,10 +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/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
 
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