소스 검색

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 년 전
부모
커밋
2d719c11cf
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  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")