diff --git a/finetune.py b/finetune.py
index 3a0ba13..84cb7d4 100644
--- a/finetune.py
+++ b/finetune.py
@@ -357,7 +357,7 @@ def train():
         model=model, tokenizer=tokenizer, args=training_args, **data_module
     )
 
-    trainer.train()
+    trainer.train(resume_from_checkpoint=training_args.resume_from_checkpoint)
     trainer.save_state()
 
     safe_save_model_for_hf_trainer(trainer=trainer, output_dir=training_args.output_dir, bias=lora_args.lora_bias)