address comments

Former-commit-id: 6bfd8442671eec4c0fb63fd34667a44760fd84e2
This commit is contained in:
Chengcheng Pei 2024-09-25 21:07:51 -07:00
parent e80c98367e
commit bc7197dcfc

View File

@ -83,8 +83,7 @@ def load_tokenizer(model_args: "ModelArguments") -> "TokenizerModule":
**init_kwargs,
)
except Exception as e:
logger.error("Failed to load tokenizer. Error: {}".format(e))
raise e
raise OSError("Failed to load tokenizer") from e
if model_args.new_special_tokens is not None:
num_added_tokens = tokenizer.add_special_tokens(