DaniilAlpha commited on
Commit
74515cd
·
1 Parent(s): a3ccca5

Update answerer.py

Browse files
Files changed (1) hide show
  1. 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