fix

flybird11111 · flybird11111 · commit c34ba4e03374 · 2025-01-17T10:42:15.000+08:00
diff --git a/colossalai/shardformer/layer/parallel_module.py b/colossalai/shardformer/layer/parallel_module.py
@@ -120,6 +120,7 @@ def _load_from_state_dict(
                         "received {}".format(key, type(input_param))
                     )
                     continue
+
                 if is_distributed_tensor(param):
                     # shard the input param
                     device_mesh = get_device_mesh(param)

Original file line number	Diff line number	Diff line change
`@@ -120,6 +120,7 @@ def _load_from_state_dict(`
`120`	`120`	`"received {}".format(key, type(input_param))`
`121`	`121`	`)`
`122`	`122`	`continue`
	`123`	`+`
`123`	`124`	`if is_distributed_tensor(param):`
`124`	`125`	`# shard the input param`
`125`	`126`	`device_mesh = get_device_mesh(param)`