diff --git a/src/llamafactory/model/loader.py b/src/llamafactory/model/loader.py index e86e7955..1fa04e0f 100644 --- a/src/llamafactory/model/loader.py +++ b/src/llamafactory/model/loader.py @@ -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(