mirror of https://github.com/hpcaitech/ColossalAI
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.cipull/5684/head
parent
2632916329
commit
88f057ce7c
|
@ -18,9 +18,9 @@ from colossalai.shardformer.layer import (
|
|||
|
||||
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
|
||||
|
||||
|
|
Loading…
Reference in New Issue