apunkt/doc-to-lora
1
0
Fork 0
mirror of https://github.com/SakanaAI/doc-to-lora.git synced 2026-07-23 17:01:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

arg name

Browse source
This commit is contained in:
51616 2025-01-04 18:24:33 +00:00
parent b9c9dc3615
commit 9d22b804b7
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
hyperlora/intx_sft.py
View file

@ -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)