DaniilAlpha commited on
Commit
e847637
·
1 Parent(s): db594e1

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -8,9 +8,9 @@ from accelerator import Accelerator
8
  from answerer import Answerer
9
  from mapper import Mapper
10
 
11
- mapper = Mapper("multi-qa-distilbert-cos-v1.bin")
12
  answerer = Answerer(
13
- model="RWKV-5-World-3B-v2-20231118-ctx16k.pth",
14
  vocab="rwkv_vocab_v20230424",
15
  strategy="cpu bf16",
16
  ctx_limit=16*1024,
 
8
  from answerer import Answerer
9
  from mapper import Mapper
10
 
11
+ mapper = Mapper("multi-qa-distilbert-cos-v1")
12
  answerer = Answerer(
13
+ model="rwkv-5-world-3b-v2-20231118-ctx16k",
14
  vocab="rwkv_vocab_v20230424",
15
  strategy="cpu bf16",
16
  ctx_limit=16*1024,