lgcharpe commited on
Commit
a36b0d0
·
verified ·
1 Parent(s): 4abdbe5

Update modeling_gptbert.py

Browse files
Files changed (1) hide show
  1. modeling_gptbert.py +2 -2
modeling_gptbert.py CHANGED
@@ -23,8 +23,8 @@ from transformers.modeling_outputs import (
23
  import math
24
  from typing import TYPE_CHECKING, Optional, Union, Tuple, List
25
 
26
- def is_flash_attn_2_available():
27
- return False
28
 
29
  if is_flash_attn_2_available():
30
  from flash_attn.flash_attn_interface import flash_attn_varlen_qkvpacked_func
 
23
  import math
24
  from typing import TYPE_CHECKING, Optional, Union, Tuple, List
25
 
26
+ # def is_flash_attn_2_available():
27
+ # return False
28
 
29
  if is_flash_attn_2_available():
30
  from flash_attn.flash_attn_interface import flash_attn_varlen_qkvpacked_func