Skip to content

Commit d972625

Browse files
Kyle1668claude
andcommitted
fix: Update eval adapter path and add PyTorch weights_only parameter
- Update custom_tasks_path in eval_adapter.py to user's local environment - Add weights_only=False to torch.load() calls in convert_neox_to_hf.py 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <[email protected]>
1 parent 7b6de79 commit d972625

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

eval_tasks/eval_adapter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -417,7 +417,7 @@ def run_eval(
417417

418418
# register all the default tasks bundled with lm-evaluation-harness repository
419419
# TODO: Set via config
420-
custom_tasks_path = "/workspace/lm_eval_tasks/"
420+
custom_tasks_path = "/home/a5k/kyleobrien.a5k/filtering_for_danger/lm_eval_tasks/"
421421
task_manager = tasks.TaskManager(include_path=custom_tasks_path)
422422
task_manager.initialize_tasks()
423423

tools/ckpts/convert_neox_to_hf.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -213,6 +213,7 @@ def load_partitions(
213213
filename_format.format(i=i),
214214
),
215215
map_location=torch.device("cuda" if torch.cuda.is_available() else "cpu"),
216+
weights_only=False,
216217
)
217218
for i in range(mp_partitions)
218219
]

0 commit comments

Comments
 (0)