P01son commited on
Commit
c611b16
1 Parent(s): c57a7ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -18,17 +18,17 @@ def init_args():
18
  global args
19
  parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
20
  args = parser.parse_args()
21
- # args.load_model_path = 'Linly-AI/ChatFlow-13B'
22
- args.load_model_path = 'Linly-AI/ChatFlow-7B'
23
  # args.load_model_path = './model_file/chatllama_7b.bin'
24
- args.config_path = './config/llama_7b.json'
25
  #args.load_model_path = './model_file/chatflow_13b.bin'
26
- # args.config_path = './config/llama_13b_config.json'
27
  args.spm_model_path = './model_file/tokenizer.model'
28
  args.batch_size = 1
29
  args.seq_length = 1024
30
  args.world_size = 1
31
- args.use_int8 = False
32
  args.top_p = 0
33
  args.repetition_penalty_range = 1024
34
  args.repetition_penalty_slope = 0
 
18
  global args
19
  parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
20
  args = parser.parse_args()
21
+ args.load_model_path = 'Linly-AI/ChatFlow-13B'
22
+ #args.load_model_path = 'Linly-AI/ChatFlow-7B'
23
  # args.load_model_path = './model_file/chatllama_7b.bin'
24
+ #args.config_path = './config/llama_7b.json'
25
  #args.load_model_path = './model_file/chatflow_13b.bin'
26
+ args.config_path = './config/llama_13b_config.json'
27
  args.spm_model_path = './model_file/tokenizer.model'
28
  args.batch_size = 1
29
  args.seq_length = 1024
30
  args.world_size = 1
31
+ args.use_int8 = True
32
  args.top_p = 0
33
  args.repetition_penalty_range = 1024
34
  args.repetition_penalty_slope = 0