improve dynamic state api

2024-05-16 13:50:48 +08:00
parent b24a18cd3a
commit d66fd89947
2 changed files with 31 additions and 12 deletions
--- a/backend-python/routes/config.py
+++ b/backend-python/routes/config.py
@@ -120,7 +120,9 @@ def update_config(body: ModelConfigBody):
        model_config = ModelConfigBody()
        global_var.set(global_var.Model_Config, model_config)
    merge_model(model_config, body)
-    exception = load_rwkv_state(global_var.get(global_var.Model), model_config.state)
+    exception = load_rwkv_state(
+        global_var.get(global_var.Model), model_config.state, True
+    )
    if exception is not None:
        raise exception
    print("Updated Model Config:", model_config)