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

icml cleaned

Browse source
This commit is contained in:
51616 2026-02-13 06:02:29 +00:00
parent c4c1ac6eaa
commit 8093817c34
30 changed files with 366 additions and 1094 deletions
Download patch file Download diff file Expand all files Collapse all files

2
data/generate_ctx_magic_number.py
View file

@ -246,7 +246,7 @@ def main():
k_values = list(range(k_start, k_end))
bin_size = len(k_values)
save_dir = f"{args.out_prefix}_{tok_bin[0]}_{tok_bin[1]}"
training_enabled = tok_bin[1] <= 512 # unchanged policy
training_enabled = tok_bin[1] <= 1024 # unchanged policy
if training_enabled:
train_data: list[dict] = []
# Distribute training budget across k values.