From bc7197dcfc45fbd97b61788c69ca9650320a2b37 Mon Sep 17 00:00:00 2001
From: Chengcheng Pei <chengchengpei@outlook.com>
Date: Wed, 25 Sep 2024 21:07:51 -0700
Subject: [PATCH] address comments

Former-commit-id: 6bfd8442671eec4c0fb63fd34667a44760fd84e2
---
 src/llamafactory/model/loader.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

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(