Crystalcareai commited on
Commit
0beaf57
·
verified ·
1 Parent(s): 3d962f0

Update modeling_quiet.py

Browse files
Files changed (1) hide show
  1. modeling_quiet.py +0 -1
modeling_quiet.py CHANGED
@@ -488,7 +488,6 @@ class QuietSdpaAttention(QuietAttention):
488
 
489
  QUIET_ATTENTION_CLASSES = {
490
  "eager": QuietAttention,
491
- "flash_attention_2": QuietFlashAttention2,
492
  "sdpa": QuietSdpaAttention,
493
  }
494
 
 
488
 
489
  QUIET_ATTENTION_CLASSES = {
490
  "eager": QuietAttention,
 
491
  "sdpa": QuietSdpaAttention,
492
  }
493