From 0fb8dbab3a889fd5d79cfe3cc4adbacf8e68531c Mon Sep 17 00:00:00 2001
From: YWMditto <862779238@qq.com>
Date: Tue, 7 Nov 2023 23:11:39 +0800
Subject: [PATCH] update InternLM/tools/load_internlm_model.py

---
 tools/load_internlm_model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tools/load_internlm_model.py b/tools/load_internlm_model.py
index 24bc39e..5cf9cda 100644
--- a/tools/load_internlm_model.py
+++ b/tools/load_internlm_model.py
@@ -169,8 +169,8 @@ def initialize_internlm_model(
         init_storage_manager(False, None, None)
     except AssertionError:
         pass
-    except Exception:
-        raise Exception
+    except Exception as e:
+        raise e
 
     model_config["dtype"] = param_dtype
     model_config["parallel_output"] = False