Update modeling_gptbert.py
Browse files- 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 |
-
|
| 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
|