Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
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/
|
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/
|
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/
|
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 |
|