fix bug in dataset (#206)

irisliu10 · web-flow · commit 0738903ff50a · 2026-01-13T16:04:33.000+08:00
diff --git 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(

Original file line number	Diff line number	Diff line change
`@@ -85,7 +85,7 @@ def __init__(`
`85`	`85`	`chat_template_type=chat_template_type,`
`86`	`86`	`display=display,`
`87`	`87`	`target_model_name_or_path=data_args.target_model_name_or_path,`
`88`		`- output_dir=data_args.output_dir,`
	`88`	`+ output_dir=getattr(data_args, "output_dir", None),`
`89`	`89`	`)`
`90`	`90`	`if data_args.training_mode == "offline":`
`91`	`91`	`self.offline_dataset_builder = DatasetBuilderFactory.create(`