Selaa lähdekoodia

fix HF export script

Eric Wang 3 vuotta sitten
vanhempi
commit
6ced8d9907
1 muutettua tiedostoa jossa 1 lisäystä ja 1 poistoa
  1. 1 1
      export_hf_checkpoint.py

+ 1 - 1
export_hf_checkpoint.py

@@ -46,7 +46,7 @@ assert not torch.allclose(first_weight_old, first_weight)
 
 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
 }