Markus28 commited on
Commit
4c4562b
1 Parent(s): 46df05d

feat: try to fix import error

Browse files
Files changed (1) hide show
  1. modeling_bert.py +5 -5
modeling_bert.py CHANGED
@@ -29,17 +29,17 @@ from transformers.models.bert.modeling_bert import (
29
  BaseModelOutputWithPoolingAndCrossAttentions,
30
  BertForPreTrainingOutput,
31
  )
32
- from .flash_components.bert_padding import (
33
  index_first_axis,
34
  index_first_axis_residual,
35
  pad_input,
36
  unpad_input,
37
  )
38
 
39
- from .flash_components.block import Block
40
- from .flash_components.embedding import BertEmbeddings
41
- from .flash_components.mha import MHA
42
- from .flash_components.mlp import FusedMLP, Mlp
43
 
44
  try:
45
  from flash_attn.ops.fused_dense import FusedDense
 
29
  BaseModelOutputWithPoolingAndCrossAttentions,
30
  BertForPreTrainingOutput,
31
  )
32
+ from flash_components.bert_padding import (
33
  index_first_axis,
34
  index_first_axis_residual,
35
  pad_input,
36
  unpad_input,
37
  )
38
 
39
+ from flash_components.block import Block
40
+ from flash_components.embedding import BertEmbeddings
41
+ from flash_components.mha import MHA
42
+ from flash_components.mlp import FusedMLP, Mlp
43
 
44
  try:
45
  from flash_attn.ops.fused_dense import FusedDense