DaniilAlpha commited on
Commit
1a5c000
·
1 Parent(s): 707a118

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -4,7 +4,7 @@ from fastapi.responses import HTMLResponse
4
  from answerer import Answerer
5
 
6
  answerer = Answerer(
7
- model="model.pth",
8
  vocab="rwkv_vocab_v20230424",
9
  strategy="cpu bf16",
10
  ctx_limit=16*1024,
 
4
  from answerer import Answerer
5
 
6
  answerer = Answerer(
7
+ model="RWKV-5-World-3B-v2-20231118-ctx16k.pth",
8
  vocab="rwkv_vocab_v20230424",
9
  strategy="cpu bf16",
10
  ctx_limit=16*1024,