Spaces:

BlinkDL
/

RWKV-Gradio-1

Running on T4

BlinkDL commited on May 11

Commit

4ae533d

•

1 Parent(s): 6cdf548

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,7 +33,7 @@ pipeline_v6 = PIPELINE(model_v6, "rwkv_vocab_v20230424")
 args = model_v6.args
 eng_name = 'rwkv-x060-eng_single_round_qa-3B-20240430-ctx1024'
-chn_name = 'rwkv-x060-chn_single_round_qa-3B-20240505-ctx1024'
 # state_eng_raw = torch.load(f'/mnt/e/RWKV-Runner/models/{eng_name}.pth', map_location=torch.device('cpu'))
 # state_chn_raw = torch.load(f'/mnt/e/RWKV-Runner/models/{chn_name}.pth', map_location=torch.device('cpu'))

 args = model_v6.args
 eng_name = 'rwkv-x060-eng_single_round_qa-3B-20240430-ctx1024'
+chn_name = 'rwkv-x060-chn_single_round_qa-3B-20240511-ctx1024'
 # state_eng_raw = torch.load(f'/mnt/e/RWKV-Runner/models/{eng_name}.pth', map_location=torch.device('cpu'))
 # state_chn_raw = torch.load(f'/mnt/e/RWKV-Runner/models/{chn_name}.pth', map_location=torch.device('cpu'))