Spaces:
Paused
Paused
Commit
·
74515cd
1
Parent(s):
a3ccca5
Update answerer.py
Browse files- answerer.py +1 -1
answerer.py
CHANGED
@@ -18,7 +18,7 @@ class Answerer:
|
|
18 |
os.environ["RWKV_JIT_ON"] = "1"
|
19 |
# os.environ["RWKV_CUDA_ON"] = "1"
|
20 |
|
21 |
-
self.__model = RWKV(hf_hub_download(repo, filename), strategy=strategy)
|
22 |
self.__pipeline = PIPELINE(self.__model, vocab)
|
23 |
self.ctx_limit = ctx_limit
|
24 |
|
|
|
18 |
os.environ["RWKV_JIT_ON"] = "1"
|
19 |
# os.environ["RWKV_CUDA_ON"] = "1"
|
20 |
|
21 |
+
self.__model = RWKV(hf_hub_download(repo, f"{filename}.pth"), strategy=strategy)
|
22 |
self.__pipeline = PIPELINE(self.__model, vocab)
|
23 |
self.ctx_limit = ctx_limit
|
24 |
|