DaniilAlpha commited on
Commit
4d1bf6e
·
1 Parent(s): 3518fdf

Update answerer.py

Browse files
Files changed (1) hide show
  1. answerer.py +2 -2
answerer.py CHANGED
@@ -1,4 +1,4 @@
1
- from typing import Dict, List
2
  import os, gc
3
  from rwkv.model import RWKV
4
  from rwkv.utils import PIPELINE, PIPELINE_ARGS
@@ -68,7 +68,7 @@ class Answerer:
68
  if "\ufffd" not in tmp:
69
  tokens.clear()
70
  result += tmp
71
- if(callback) callback(result)
72
 
73
  tokens.clear()
74
  occurrences.clear()
 
1
+ from typing import Callable, Dict, List
2
  import os, gc
3
  from rwkv.model import RWKV
4
  from rwkv.utils import PIPELINE, PIPELINE_ARGS
 
68
  if "\ufffd" not in tmp:
69
  tokens.clear()
70
  result += tmp
71
+ if(callback): callback(result)
72
 
73
  tokens.clear()
74
  occurrences.clear()