openbmb
/

BitCPM-CANN-0.5B

Text Generation

Model card Files Files and versions

guanwenyu1995 commited on 11 days ago

Commit

929dac8

·

verified ·

1 Parent(s): e248676

Update modeling_minicpm.py

Files changed (1) hide show

modeling_minicpm.py +1 -1

modeling_minicpm.py CHANGED Viewed

@@ -24,7 +24,7 @@ import torch.utils.checkpoint
 from torch import nn
 from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
 from transformers.activations import ACT2FN
-from transformers.cache_utils import Cache, DynamicCache, CacheLayerMixin, DynamicLayer
 from transformers.modeling_attn_mask_utils import (
     AttentionMaskConverter,
     _prepare_4d_attention_mask,

 from torch import nn
 from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
 from transformers.activations import ACT2FN
+from transformers.cache_utils import Cache, DynamicCache
 from transformers.modeling_attn_mask_utils import (
     AttentionMaskConverter,
     _prepare_4d_attention_mask,