From a3bc2bc14cc5be8ea7ef0c87f96eefeae4c4869a Mon Sep 17 00:00:00 2001 From: CjhHa1 Date: Mon, 11 Sep 2023 16:57:14 +0800 Subject: [PATCH] revise infer default --- colossalai/shardformer/shard/shard_config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/colossalai/shardformer/shard/shard_config.py b/colossalai/shardformer/shard/shard_config.py index 7e38255c4822..4380ac30814d 100644 --- a/colossalai/shardformer/shard/shard_config.py +++ b/colossalai/shardformer/shard/shard_config.py @@ -76,4 +76,4 @@ def _infer(self): """ Set default params for inference. """ - self.pipeline_stage_manager = None + assert self.pipeline_stage_manager is None, "pipeline parallelism is not supported in inference for now"