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

toy ctx magic num eval upto 32k + remove configs

Browse source
This commit is contained in:
51616 2025-09-01 05:57:06 +00:00
parent a269aff51d
commit 3505b006ad
14 changed files with 14 additions and 200 deletions
Download patch file Download diff file Expand all files Collapse all files

1
data/generate_ctx_magic_number.py
View file

@ -168,6 +168,7 @@ def main():
tok_bins = [(32, 128), (128, 256), (256, 512), (512, 1024), (32, 1024)] + [
(1024 * i, 1024 * (i + 1)) for i in range(1, 16)
]
tok_bins += [(2**14 + 2**12 * (i), 2**14 + 2**12 * (i + 1)) for i in range(4)]
if args.only_first_n_bins is not None:
tok_bins = tok_bins[: args.only_first_n_bins]