[pre-commit.ci] auto fixes from pre-commit.com hooks

pre-commit-ci[bot] · pre-commit-ci[bot] · commit 88f057ce7c52 · 2024-05-07T07:03:47.000Z
for more information, see https://pre-commit.ci
diff --git a/colossalai/shardformer/policies/mistral.py b/colossalai/shardformer/policies/mistral.py
@@ -18,9 +18,9 @@
 
 from ..modeling.mistral import (
     MistralForwards,
+    get_lm_forward_with_dist_cross_entropy,
     get_mistral_flash_attention_forward,
     get_mistral_model_forward_for_flash_attn,
-    get_lm_forward_with_dist_cross_entropy,
 )
 from .base_policy import ModulePolicyDescription, Policy, SubModuleReplacementDescription
 

Original file line number	Diff line number	Diff line change
`@@ -18,9 +18,9 @@`
`18`	`18`
`19`	`19`	`from ..modeling.mistral import (`
`20`	`20`	`MistralForwards,`
	`21`	`+ get_lm_forward_with_dist_cross_entropy,`
`21`	`22`	`get_mistral_flash_attention_forward,`
`22`	`23`	`get_mistral_model_forward_for_flash_attn,`
`23`		`- get_lm_forward_with_dist_cross_entropy,`
`24`	`24`	`)`
`25`	`25`	`from .base_policy import ModulePolicyDescription, Policy, SubModuleReplacementDescription`
`26`	`26`