NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-24 17:04:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

linting

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-02-03 12:23:34 -05:00
parent e932369777
commit 8fabb701f8
2 changed files with 39 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
example_trainer/data.py
View file

@ -249,9 +249,9 @@ def get_data(
sample_lp = sample_item["inference_logprobs"][0] if sample_item["inference_logprobs"] else []
print(f" [Data] ✓ inference_logprobs found in batch (sample len: {len(sample_lp)})")
else:
print(f" [Data] ⚠ inference_logprobs key exists but is empty!")
print(" [Data] ⚠ inference_logprobs key exists but is empty!")
else:
print(f" [Data] ⚠ NO inference_logprobs in batch data!")
print(" [Data] ⚠ NO inference_logprobs in batch data!")
print(f" [Data] Keys in first item: {list(data['batch'][0].keys())}")
_logged_logprob_warning = True