RWKV-Red-Team
/

Mini_RWKV_7_34.2M

Model card Files Files and versions

Alic-Li commited on Jul 18

Commit

10dad5c

·

verified ·

1 Parent(s): 2dd2cd9

Update API_DEMO_CHAT.py

Files changed (1) hide show

API_DEMO_CHAT.py +1 -1

API_DEMO_CHAT.py CHANGED Viewed

@@ -26,7 +26,7 @@ args = types.SimpleNamespace()
 args.strategy = "cuda fp16"  # use CUDA, fp16
-args.MODEL_NAME = "/home/alic-li/rwkv7-g1-tanslate/rwkv-final-sft-512"
 ########################################################################################################

 args.strategy = "cuda fp16"  # use CUDA, fp16
+args.MODEL_NAME = "./rwkv-final-sft-2048.pth"
 ########################################################################################################