Update API_DEMO_CHAT.py
Browse files- API_DEMO_CHAT.py +1 -1
API_DEMO_CHAT.py
CHANGED
|
@@ -26,7 +26,7 @@ args = types.SimpleNamespace()
|
|
| 26 |
|
| 27 |
args.strategy = "cuda fp16" # use CUDA, fp16
|
| 28 |
|
| 29 |
-
args.MODEL_NAME = "
|
| 30 |
|
| 31 |
|
| 32 |
########################################################################################################
|
|
|
|
| 26 |
|
| 27 |
args.strategy = "cuda fp16" # use CUDA, fp16
|
| 28 |
|
| 29 |
+
args.MODEL_NAME = "./rwkv-final-sft-2048.pth"
|
| 30 |
|
| 31 |
|
| 32 |
########################################################################################################
|