Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -10,10 +10,10 @@ HAS_GPU = False
|
|
10 |
# Model title and context size limit
|
11 |
ctx_limit = 4000
|
12 |
title = "1B5"
|
13 |
-
model_file = "RWKV-5-World-1B5-v2-20231025-ctx4096"
|
14 |
|
15 |
#title = "RWKV-4-World"
|
16 |
-
|
17 |
|
18 |
# Get the GPU count
|
19 |
try:
|
@@ -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 |
-
|
44 |
model = RWKV(model=model_path, strategy=MODEL_STRAT)
|
45 |
from rwkv.utils import PIPELINE, PIPELINE_ARGS
|
46 |
pipeline = PIPELINE(model, "rwkv_vocab_v20230424")
|
|
|
10 |
# Model title and context size limit
|
11 |
ctx_limit = 4000
|
12 |
title = "1B5"
|
13 |
+
#model_file = "RWKV-5-World-1B5-v2-20231025-ctx4096"
|
14 |
|
15 |
#title = "RWKV-4-World"
|
16 |
+
model_file = "rwkv-4-world/RWKV-4-World-CHNtuned-1.5B-v1-20230620-ctx4096.pth"
|
17 |
|
18 |
# Get the GPU count
|
19 |
try:
|
|
|
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")
|