Skip to content
代码片段 群组 项目
未验证 提交 8e14344e 编辑于 作者: Jiarui Fang's avatar Jiarui Fang 提交者: GitHub
浏览文件

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

上级 05545bfe
无相关合并请求
......@@ -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.
......
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册