Spaces:
Running
on
Zero
Running
on
Zero
alex
commited on
Commit
·
6ffc50e
1
Parent(s):
b2ed9cf
debug
Browse files- ovi/modules/attention.py +1 -0
ovi/modules/attention.py
CHANGED
|
@@ -4,6 +4,7 @@ import torch
|
|
| 4 |
try:
|
| 5 |
import flash_attn_interface
|
| 6 |
FLASH_ATTN_3_AVAILABLE = True
|
|
|
|
| 7 |
except ModuleNotFoundError:
|
| 8 |
FLASH_ATTN_3_AVAILABLE = False
|
| 9 |
|
|
|
|
| 4 |
try:
|
| 5 |
import flash_attn_interface
|
| 6 |
FLASH_ATTN_3_AVAILABLE = True
|
| 7 |
+
print(f'FLASH_ATTN_3_AVAILABLE:{FLASH_ATTN_3_AVAILABLE}')
|
| 8 |
except ModuleNotFoundError:
|
| 9 |
FLASH_ATTN_3_AVAILABLE = False
|
| 10 |
|