icashwave commited on
Commit
815dd65
1 Parent(s): f1a7d6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -39,8 +39,8 @@ if HAS_GPU == True :
39
 
40
  # Load the model accordingly
41
  from rwkv.model import RWKV
42
- #model_path = hf_hub_download(repo_id="BlinkDL/rwkv-5-world", filename=f"{model_file}.pth")
43
- model_path = hf_hub_download(repo_id="BlinkDL/rwkv-4-world", filename=f"{model_file}.pth")
44
  model = RWKV(model=model_path, strategy=MODEL_STRAT)
45
  from rwkv.utils import PIPELINE, PIPELINE_ARGS
46
  pipeline = PIPELINE(model, "rwkv_vocab_v20230424")
 
39
 
40
  # Load the model accordingly
41
  from rwkv.model import RWKV
42
+ model_path = hf_hub_download(repo_id="BlinkDL/rwkv-5-world", filename=f"{model_file}.pth")
43
+ #model_path = hf_hub_download(repo_id="BlinkDL/rwkv-4-world", filename=f"{model_file}.pth")
44
  model = RWKV(model=model_path, strategy=MODEL_STRAT)
45
  from rwkv.utils import PIPELINE, PIPELINE_ARGS
46
  pipeline = PIPELINE(model, "rwkv_vocab_v20230424")