Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,7 +24,7 @@ gen_limit = 1000
|
|
| 24 |
########################## text rwkv ################################################################
|
| 25 |
from rwkv.utils import PIPELINE, PIPELINE_ARGS
|
| 26 |
|
| 27 |
-
title_v6 = "rwkv7-
|
| 28 |
model_path_v6 = hf_hub_download(repo_id="BlinkDL/rwkv7-g1", filename=f"{title_v6}.pth")
|
| 29 |
model_v6 = RWKV(model=model_path_v6.replace('.pth',''), strategy='cuda fp16')
|
| 30 |
pipeline_v6 = PIPELINE(model_v6, "rwkv_vocab_v20230424")
|
|
|
|
| 24 |
########################## text rwkv ################################################################
|
| 25 |
from rwkv.utils import PIPELINE, PIPELINE_ARGS
|
| 26 |
|
| 27 |
+
title_v6 = "rwkv7-g1a4-2.9b-20251118-ctx8192"
|
| 28 |
model_path_v6 = hf_hub_download(repo_id="BlinkDL/rwkv7-g1", filename=f"{title_v6}.pth")
|
| 29 |
model_v6 = RWKV(model=model_path_v6.replace('.pth',''), strategy='cuda fp16')
|
| 30 |
pipeline_v6 = PIPELINE(model_v6, "rwkv_vocab_v20230424")
|