[hotfix] fix a type in ColoInitContext (#2106)

8e14344e · Jiarui Fang · GitHub · 05545bfe · 8e14344e
Unverified Commit 8e14344e authored Dec 09, 2022 by Jiarui Fang Committed by GitHub Dec 09, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

colossalai/utils/model/colo_init_context.py colossalai/utils/model/colo_init_context.py +1 -1

No files found.
--- a/colossalai/utils/model/colo_init_context.py
+++ b/colossalai/utils/model/colo_init_context.py
@@ -38,7 +38,7 @@ def _convert_to_coloparam(param: torch.nn.Parameter,
    requires_grad = param.requires_grad

    if param.device.type == 'meta':
-        raise NotImplemented(
+        raise NotImplementedError(
            "ColoInitContext is initializing a model with meta parameters! This is not allowed right now!")
    else:
        # param is the global tensor.