From 2a8d6a356a8eb4077a3c4aadb0f4728110ca21cf Mon Sep 17 00:00:00 2001 From: root Date: Tue, 13 Jan 2026 15:47:36 +0800 Subject: [PATCH] fix bug in dataset --- angelslim/compressor/speculative/train/data/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/angelslim/compressor/speculative/train/data/dataset.py b/angelslim/compressor/speculative/train/data/dataset.py index 12eeabac..722f0d15 100644 --- a/angelslim/compressor/speculative/train/data/dataset.py +++ b/angelslim/compressor/speculative/train/data/dataset.py @@ -85,7 +85,7 @@ def __init__( chat_template_type=chat_template_type, display=display, target_model_name_or_path=data_args.target_model_name_or_path, - output_dir=data_args.output_dir, + output_dir=getattr(data_args, "output_dir", None), ) if data_args.training_mode == "offline": self.offline_dataset_builder = DatasetBuilderFactory.create(