address comments

Former-commit-id: 6311bb2ca266ce156537cfa477202b2904921593
This commit is contained in:
Chengcheng Pei 2024-09-25 21:07:51 -07:00
parent 26bbfc084d
commit cacafb0038

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(