update InternLM/tools/load_internlm_model.py

2023-11-07 23:11:39 +08:00 · 2023-11-07 23:11:39 +08:00 · 0fb8dbab3a
parent 1a97169fab
commit 0fb8dbab3a
1 changed files with 2 additions and 2 deletions
--- 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