huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 20:34:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

add back warning

Browse Source
This commit is contained in:
fxmarty 2024-05-17 16:51:31 +00:00
parent c6565e8259
commit f82ae76dff
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
server/text_generation_server/models/__init__.py
View File

@ -84,6 +84,7 @@ try:
HAS_FLASH_ATTN_V2_ROCM, HAS_FLASH_ATTN_V2_ROCM,
) )
except ImportError as e: except ImportError as e:
logger.warning(f"Could not import Flash Attention enabled models: {e}")
FLASH_ATTENTION = False FLASH_ATTENTION = False
HAS_FLASH_ATTN_V2_CUDA = False HAS_FLASH_ATTN_V2_CUDA = False
HAS_FLASH_ATTN_V2_ROCM = False HAS_FLASH_ATTN_V2_ROCM = False