Browse Source

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 3 năm trước cách đây
mục cha
commit
2d719c11cf
1 tập tin đã thay đổi với 1 bổ sung1 xóa
  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")