AoiKazama
/

Kinoe-7B

Text Generation

Model card Files Files and versions Community

AoiKazama commited on May 27

Commit

24354f1

•

1 Parent(s): 2d5b127

FIX

Files changed (1) hide show

modeling_kinoe.py +1 -1

modeling_kinoe.py CHANGED Viewed

@@ -31,7 +31,7 @@ from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
 from transformers.activations import ACT2FN
 from .cache_utils import Cache, DynamicCache, SlidingWindowCache, StaticCache
-from transformers.modeling_attn_mask_utils import AttentionMaskConverter
 from transformers.modeling_outputs import (
     BaseModelOutputWithPast,
     CausalLMOutputWithPast,

 from transformers.activations import ACT2FN
 from .cache_utils import Cache, DynamicCache, SlidingWindowCache, StaticCache
+from .modeling_attn_mask_utils import AttentionMaskConverter
 from transformers.modeling_outputs import (
     BaseModelOutputWithPast,
     CausalLMOutputWithPast,