fix(train): restore save_model, accidentally dropped in wandb PR

hBouanane · hBouanane · commit 5013c7a1fda2 · 2026-07-06T23:53:10.000+01:00
diff --git a/src/darija_translator/train.py b/src/darija_translator/train.py
@@ -41,3 +41,8 @@ def build_trainer(model, tokenizer, train_dataset, eval_dataset,
         instruction_part="<|im_start|>user\n",
         response_part="<|im_start|>assistant\n",
     )
+
+
+def save_model(model, tokenizer, config: TrainConfig):
+    model.save_pretrained(config.output_dir)
+    tokenizer.save_pretrained(config.output_dir)

Original file line number	Diff line number	Diff line change
`@@ -41,3 +41,8 @@ def build_trainer(model, tokenizer, train_dataset, eval_dataset,`
`41`	`41`	`instruction_part="<\|im_start\|>user\n",`
`42`	`42`	`response_part="<\|im_start\|>assistant\n",`
`43`	`43`	`)`
	`44`	`+`
	`45`	`+`
	`46`	`+def save_model(model, tokenizer, config: TrainConfig):`
	`47`	`+ model.save_pretrained(config.output_dir)`
	`48`	`+ tokenizer.save_pretrained(config.output_dir)`