update state-tuned safetensors converter

2024-05-03 13:10:49 +08:00 · 2024-05-03 13:10:49 +08:00 · 150bb089cf
commit 150bb089cf
parent 5c8a637cf5
1 changed files with 1 additions and 0 deletions
--- a/backend-python/convert_safetensors.py
+++ b/backend-python/convert_safetensors.py
@ -102,6 +102,7 @@ if __name__ == "__main__":
                "time_mix_w2",
                "time_decay_w1",
                "time_decay_w2",
+                "time_state",
                "lora.0",
            ],
        )