tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2025-12-06 12:22:18 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix runcpu when num_fewshot is greater than data

Browse Source
This commit is contained in:
burtenshaw 2025-10-21 19:37:46 +02:00
parent 50bea28ef9
commit cc8d819286
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nanochat/core_eval.py
View File

@ -174,7 +174,7 @@ def evaluate_example(idx, model, tokenizer, data, device, task_meta):
# Sample few-shot examples (excluding current item) # Sample few-shot examples (excluding current item)
fewshot_examples = [] fewshot_examples = []
if num_fewshot > 0: if num_fewshot > 0 and len(data) > num_fewshot:
rng = random.Random(1234 + idx) rng = random.Random(1234 + idx)
available_indices = [i for i in range(len(data)) if i != idx] available_indices = [i for i in range(len(data)) if i != idx]
fewshot_indices = rng.sample(available_indices, num_fewshot) fewshot_indices = rng.sample(available_indices, num_fewshot)