P01son commited on
Commit
657102d
·
1 Parent(s): c611b16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ def init_model():
46
  model = LLaMa(args)
47
  torch.set_default_tensor_type(torch.FloatTensor)
48
  # args.load_model_path = hf_hub_download(repo_id=args.load_model_path, filename='chatflow_13b.bin')
49
- args.load_model_path = hf_hub_download(repo_id=args.load_model_path, filename='chatflow_7b.bin')
50
  model = load_model(model, args.load_model_path)
51
  model.eval()
52
 
 
46
  model = LLaMa(args)
47
  torch.set_default_tensor_type(torch.FloatTensor)
48
  # args.load_model_path = hf_hub_download(repo_id=args.load_model_path, filename='chatflow_13b.bin')
49
+ args.load_model_path = hf_hub_download(repo_id=args.load_model_path, filename='chatflow_13b.bin')
50
  model = load_model(model, args.load_model_path)
51
  model.eval()
52