DaniilAlpha commited on
Commit
201400d
·
1 Parent(s): 2a39db4

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -3
main.py CHANGED
@@ -4,8 +4,7 @@ from fastapi.responses import HTMLResponse
4
  from answerer import Answerer
5
 
6
  answerer = Answerer(
7
- repo="BlinkDL/rwkv-5-world",
8
- model="RWKV-5-World-3B-v2-20231118-ctx16k",
9
  vocab="rwkv_vocab_v20230424",
10
  strategy="cpu bf16",
11
  ctx_limit=16*1024,
@@ -44,7 +43,6 @@ HTML = """
44
  }
45
 
46
  function answer(value) {
47
- alert(value);
48
  output.innerHTML = value;
49
  }
50
  </script>
 
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,
 
43
  }
44
 
45
  function answer(value) {
 
46
  output.innerHTML = value;
47
  }
48
  </script>