Skip to content

GitLab

Unverified Commit 8e14344e authored by Jiarui Fang's avatar Jiarui Fang Committed by GitHub
Browse files

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

parent 05545bfe
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
colossalai/utils/model/colo_init_context.py
View file @ 8e14344e
...@@ -38,7 +38,7 @@ def _convert_to_coloparam(param: torch.nn.Parameter, ...@@ -38,7 +38,7 @@ def _convert_to_coloparam(param: torch.nn.Parameter,
requires_grad = param.requires_grad requires_grad = param.requires_grad
if param.device.type == 'meta': if param.device.type == 'meta':
raise NotImplemented( raise NotImplementedError(
"ColoInitContext is initializing a model with meta parameters! This is not allowed right now!") "ColoInitContext is initializing a model with meta parameters! This is not allowed right now!")
else: else:
# param is the global tensor. # param is the global tensor.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment