From 9d22b804b746f2ee55aef55c3e5d0923739a0752 Mon Sep 17 00:00:00 2001 From: 51616 Date: Sat, 4 Jan 2025 18:24:33 +0000 Subject: [PATCH] arg name --- hyperlora/intx_sft.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hyperlora/intx_sft.py b/hyperlora/intx_sft.py index 665258d..c34926e 100644 --- a/hyperlora/intx_sft.py +++ b/hyperlora/intx_sft.py @@ -391,7 +391,7 @@ def main(output_dir): if "validation" in tokenized_ds: val_ds["validation"] = tokenized_ds["validation"] val_ds_size = len(val_ds["validation"]) - val_indices = np.random.permutation(val_ds_size)[: args.max_val_samples] + val_indices = np.random.permutation(val_ds_size)[: data_args.max_val_samples] val_ds["validation"] = val_ds["validation"].select(val_indices) test_ds = tokenized_ds.get("test", None)