BlinkDL commited on
Commit
c980909
1 Parent(s): 9fed525

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -25,17 +25,17 @@ pipeline = PIPELINE(model, "rwkv_vocab_v20230424")
25
 
26
  args = model.args
27
  eng_name = 'rwkv6-world-v3-7b-eng_QA-20241114-ctx2048'
28
- eng_file = hf_hub_download(repo_id="BlinkDL/temp-latest-training-models", filename=f"states/{eng_name}.pth")
29
  state_eng_raw = torch.load(eng_file)
30
  state_eng = [None] * args.n_layer * 3
31
 
32
  chn_name = 'rwkv6-world-v3-7b-chn_问答QA-20241114-ctx2048'
33
- chn_file = hf_hub_download(repo_id="BlinkDL/temp-latest-training-models", filename=f"states/{chn_name}.pth")
34
  state_chn_raw = torch.load(chn_file)
35
  state_chn = [None] * args.n_layer * 3
36
 
37
  wyw_name = 'rwkv6-world-v3-7b-chn_文言文QA-20241114-ctx2048'
38
- wyw_file = hf_hub_download(repo_id="BlinkDL/temp-latest-training-models", filename=f"states/{wyw_name}.pth")
39
  state_wyw_raw = torch.load(wyw_file)
40
  state_wyw = [None] * args.n_layer * 3
41
 
 
25
 
26
  args = model.args
27
  eng_name = 'rwkv6-world-v3-7b-eng_QA-20241114-ctx2048'
28
+ eng_file = hf_hub_download(repo_id="BlinkDL/rwkv-6-misc", filename=f"states/{eng_name}.pth")
29
  state_eng_raw = torch.load(eng_file)
30
  state_eng = [None] * args.n_layer * 3
31
 
32
  chn_name = 'rwkv6-world-v3-7b-chn_问答QA-20241114-ctx2048'
33
+ chn_file = hf_hub_download(repo_id="BlinkDL/rwkv-6-misc", filename=f"states/{chn_name}.pth")
34
  state_chn_raw = torch.load(chn_file)
35
  state_chn = [None] * args.n_layer * 3
36
 
37
  wyw_name = 'rwkv6-world-v3-7b-chn_文言文QA-20241114-ctx2048'
38
+ wyw_file = hf_hub_download(repo_id="BlinkDL/rwkv-6-misc", filename=f"states/{wyw_name}.pth")
39
  state_wyw_raw = torch.load(wyw_file)
40
  state_wyw = [None] * args.n_layer * 3
41