csuhan commited on
Commit
5b036b0
1 Parent(s): 0a39676

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -423,7 +423,7 @@ class DemoConfig:
423
  llama_type = "onellm"
424
  llama_config = "config/llama2/7B.json"
425
  model_max_seq_len = 2048
426
- pretrained_path = "csuhan/OneLLM-7B-hf-v1.1"
427
  # pretrained_path = "/home/pgao/jiaming/weights/7B_v20_splits/"
428
  ckpt_format = "consolidated.00-of-01.s{}.pth"
429
  num_ckpts = 10
 
423
  llama_type = "onellm"
424
  llama_config = "config/llama2/7B.json"
425
  model_max_seq_len = 2048
426
+ pretrained_path = "csuhan/OneLLM-7B-hf"
427
  # pretrained_path = "/home/pgao/jiaming/weights/7B_v20_splits/"
428
  ckpt_format = "consolidated.00-of-01.s{}.pth"
429
  num_ckpts = 10