Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -39,7 +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 = RWKV(model=model_path, strategy=MODEL_STRAT)
|
44 |
from rwkv.utils import PIPELINE, PIPELINE_ARGS
|
45 |
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")
|