Spaces:
Paused
Paused
Commit
·
79d23f5
1
Parent(s):
01ce711
Update answerer.py
Browse files- answerer.py +1 -1
answerer.py
CHANGED
|
@@ -9,7 +9,7 @@ class Answerer:
|
|
| 9 |
os.environ["RWKV_JIT_ON"] = "1"
|
| 10 |
# os.environ["RWKV_CUDA_ON"] = "1"
|
| 11 |
|
| 12 |
-
self.__model = RWKV(model, strategy=strategy)
|
| 13 |
self.__pipeline = PIPELINE(self.__model, vocab)
|
| 14 |
self.ctx_limit = ctx_limit
|
| 15 |
|
|
|
|
| 9 |
os.environ["RWKV_JIT_ON"] = "1"
|
| 10 |
# os.environ["RWKV_CUDA_ON"] = "1"
|
| 11 |
|
| 12 |
+
self.__model = RWKV(f"models/{model}.pth", strategy=strategy)
|
| 13 |
self.__pipeline = PIPELINE(self.__model, vocab)
|
| 14 |
self.ctx_limit = ctx_limit
|
| 15 |
|