update state-tuned safetensors converter

josStorer · May 3, 2024 · 150bb08 · 150bb08
1 parent 5c8a637
commit 150bb08
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/backend-python/convert_safetensors.py b/backend-python/convert_safetensors.py
@@ -102,6 +102,7 @@ def convert_file(pt_filename: str, sf_filename: str, rename={}, transpose_names=
                 "time_mix_w2",
                 "time_decay_w1",
                 "time_decay_w2",
+                "time_state",
                 "lora.0",
             ],
         )