DaniilAlpha commited on
Commit
6f601b3
·
1 Parent(s): 1519720

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -1
main.py CHANGED
@@ -8,7 +8,9 @@ from accelerator import Accelerator
8
  from answerer import Answerer
9
  from mapper import Mapper
10
 
11
- # mapper = Mapper("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
 
 
12
  answerer = Answerer(
13
  model="RWKV-5-World-3B-v2-20231118-ctx16k",
14
  vocab="rwkv_vocab_v20230424",
 
8
  from answerer import Answerer
9
  from mapper import Mapper
10
 
11
+ try: mapper = Mapper("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
12
+ except: print("ERROR! cannot load Mapper model!")
13
+
14
  answerer = Answerer(
15
  model="RWKV-5-World-3B-v2-20231118-ctx16k",
16
  vocab="rwkv_vocab_v20230424",