From 8e14344ec9815a4899c19ce1618f154a395875d4 Mon Sep 17 00:00:00 2001
From: Jiarui Fang <fangjiarui123@gmail.com>
Date: Fri, 9 Dec 2022 11:44:39 +0800
Subject: [PATCH] [hotfix] fix a type in ColoInitContext (#2106)

---
 colossalai/utils/model/colo_init_context.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/colossalai/utils/model/colo_init_context.py b/colossalai/utils/model/colo_init_context.py
index 851543e4a..6cb885321 100644
--- 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.