Spaces:
Paused
Paused
Commit
·
d9dbfd0
1
Parent(s):
f6a9905
Update main.py
Browse files
main.py
CHANGED
|
@@ -8,7 +8,7 @@ from accelerator import Accelerator
|
|
| 8 |
from answerer import Answerer
|
| 9 |
from mapper import Mapper
|
| 10 |
|
| 11 |
-
|
| 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 |
+
mapper = Mapper("multi-qa-distilbert-cos-v1")
|
| 12 |
answerer = Answerer(
|
| 13 |
model="RWKV-5-World-3B-v2-20231118-ctx16k",
|
| 14 |
vocab="rwkv_vocab_v20230424",
|