From 1b4da65e37dba14c87562a3b61c3f28e1522fb6c Mon Sep 17 00:00:00 2001 From: Jena Hwang Date: Mon, 16 Dec 2024 08:42:21 -0800 Subject: [PATCH] fix --- scripts/convert_checkpoints_batch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/convert_checkpoints_batch.py b/scripts/convert_checkpoints_batch.py index 0075211bc..1ea5fa688 100644 --- a/scripts/convert_checkpoints_batch.py +++ b/scripts/convert_checkpoints_batch.py @@ -94,7 +94,7 @@ def convert_checkpoint(cps, load_dir="/data/input", sanity_check=False, weka_pre if not path_found: conversion_status = "new" converted_path = local_path - conversion_cmd = f"python hf_olmo/convert_olmo_to_hf.py --checkpoint-dir '{checkpoint_path}' --destination-dir '{local_path}' --tokenizer 'allenai/gpt-neox-olmo-dolma-v1_5' --cleanup-local-dir" + conversion_cmd = f"python hf_olmo/convert_olmo_to_hf.py --checkpoint-dir '{checkpoint_path}' --destination-dir '{local_path}' --tokenizer 'allenai/gpt-neox-olmo-dolma-v1_5'" if sanity_check: print("SANITY CHECK MODE (not running the conversion)")