fix HF export script

cheerml · Mar 18, 2023 · 6ced8d9 · 6ced8d9
1 parent 8dc0f61
commit 6ced8d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/export_hf_checkpoint.py b/export_hf_checkpoint.py
@@ -46,7 +46,7 @@
 
 lora_model_sd = lora_model.state_dict()
 deloreanized_sd = {
-    k.replace("base_model.model.model", "model"): v
+    k.replace("base_model.model.", ""): v
     for k, v in lora_model_sd.items()
     if "lora" not in k
 }