reed
commited on
Commit
•
49d999c
1
Parent(s):
86ebcde
add version info
Browse files- modeling_yi.py +1 -1
modeling_yi.py
CHANGED
|
@@ -32,7 +32,7 @@ except Exception:
|
|
| 32 |
if is_flash_attn_available:
|
| 33 |
from flash_attn import __version__
|
| 34 |
|
| 35 |
-
assert __version__ >= "2.3.0", "please update your flash_attn version"
|
| 36 |
|
| 37 |
logger = logging.get_logger(__name__)
|
| 38 |
|
|
|
|
| 32 |
if is_flash_attn_available:
|
| 33 |
from flash_attn import __version__
|
| 34 |
|
| 35 |
+
assert __version__ >= "2.3.0", "please update your flash_attn version (>= 2.3.0)"
|
| 36 |
|
| 37 |
logger = logging.get_logger(__name__)
|
| 38 |
|