From 5ca623a166037b4fc79c2a05c11a51cb21808cb8 Mon Sep 17 00:00:00 2001 From: Manan17 Date: Wed, 11 Mar 2026 00:58:22 +0000 Subject: [PATCH] fixing gguf export for gemma with text --- studio/backend/core/training/trainer.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/studio/backend/core/training/trainer.py b/studio/backend/core/training/trainer.py index 2e37cad189..1f2fedeb7c 100644 --- a/studio/backend/core/training/trainer.py +++ b/studio/backend/core/training/trainer.py @@ -2564,6 +2564,10 @@ class UnslothTrainer: if eval_dataset is not None: trainer_kwargs["eval_dataset"] = eval_dataset self.trainer = SFTTrainer(**trainer_kwargs) + # Restore the full processor as processing_class so checkpoint + # saves include preprocessor_config.json (needed for GGUF export). + if sft_tokenizer is not self.tokenizer: + self.trainer.processing_class = self.tokenizer print("Trainer initialized\n") # ========== TRAIN ON RESPONSES ONLY ==========