Procházet zdrojové kódy

Suzhiba/fix resume from checkpoint bug (#322)

* fix bug when use resume_from_checkpoint to resume lora params

* update

---------

Co-authored-by: 苏志霸 <[email protected]>
suzhiba před 3 roky
rodič
revize
2d719c11cf
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      finetune.py

+ 1 - 1
finetune.py

@@ -202,7 +202,7 @@ def train(
         if os.path.exists(checkpoint_name):
             print(f"Restarting from {checkpoint_name}")
             adapters_weights = torch.load(checkpoint_name)
-            model = set_peft_model_state_dict(model, adapters_weights)
+            set_peft_model_state_dict(model, adapters_weights)
         else:
             print(f"Checkpoint {checkpoint_name} not found")