katuni4ka commited on
Commit
9fc8a74
·
verified ·
1 Parent(s): edc7c55

Update modeling_minicpm.py

Browse files
Files changed (1) hide show
  1. modeling_minicpm.py +1 -1
modeling_minicpm.py CHANGED
@@ -755,7 +755,7 @@ class MiniCPMSdpaAttention(MiniCPMAttention):
755
 
756
  MINICPM_ATTENTION_CLASSES = {
757
  "eager": MiniCPMAttention,
758
- #"flash_attention_2": MiniCPMFlashAttention2,
759
  "sdpa": MiniCPMSdpaAttention,
760
  }
761
 
 
755
 
756
  MINICPM_ATTENTION_CLASSES = {
757
  "eager": MiniCPMAttention,
758
+ #"flash_attention_2": MiniCPMFlashAttention2,
759
  "sdpa": MiniCPMSdpaAttention,
760
  }
761