johnpaulbin commited on
Commit
4e71fd3
·
1 Parent(s): eb4b888

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ os.environ["RWKV_CUDA_ON"] = '0' # if '1' then use CUDA kernel for seq mode (muc
10
 
11
  from rwkv.model import RWKV
12
  model_path = hf_hub_download(repo_id="BlinkDL/rwkv-4-world", filename=f"{title}.pth")
13
- model = RWKV(model=model_path, strategy='cpu')
14
  from rwkv.utils import PIPELINE, PIPELINE_ARGS
15
  pipeline = PIPELINE(model, "rwkv_vocab_v20230424")
16
 
 
10
 
11
  from rwkv.model import RWKV
12
  model_path = hf_hub_download(repo_id="BlinkDL/rwkv-4-world", filename=f"{title}.pth")
13
+ model = RWKV(model=model_path, strategy='cpu fp32')
14
  from rwkv.utils import PIPELINE, PIPELINE_ARGS
15
  pipeline = PIPELINE(model, "rwkv_vocab_v20230424")
16