github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-27 12:43:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix typo change dosen't to doesn't (#5308)

Browse Source
This commit is contained in:
digger yu
2024-01-30 09:57:38 +08:00
committed by GitHub
parent 6a3086a505
commit 71321a07cf
8 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
colossalai/shardformer/policies/mistral.py
View File

@@ -35,7 +35,7 @@ class MistralPolicy(Policy):
if self.shard_config.enable_sequence_parallelism:
self.shard_config.enable_sequence_parallelism = False
warnings.warn(
"Mistral dosen't support sequence parallelism now, will ignore the sequence parallelism flag."
"Mistral doesn't support sequence parallelism now, will ignore the sequence parallelism flag."
)
if self.shard_config.enable_tensor_parallelism:
@@ -136,7 +136,7 @@ class MistralModelPolicy(MistralPolicy):
def module_policy(self):
if self.pipeline_stage_manager:
warnings.warn("Mistral dosen't support pipeline parallelism now.")
warnings.warn("Mistral doesn't support pipeline parallelism now.")
return super().module_policy()
@@ -160,7 +160,7 @@ class MistralForCausalLMPolicy(MistralPolicy):
}
if self.pipeline_stage_manager:
warnings.warn("Mistral dosen't support pipeline parallelism now.")
warnings.warn("Mistral doesn't support pipeline parallelism now.")
policy.update(new_item)
@@ -186,7 +186,7 @@ class MistralForSequenceClassificationPolicy(MistralPolicy):
}
if self.pipeline_stage_manager:
warnings.warn("Mistral dosen't support pipeline parallelism now.")
warnings.warn("Mistral doesn't support pipeline parallelism now.")
policy.update(new_item)
return policy