Update to new version of base image

This commit is contained in:
2025-10-24 15:45:06 +08:00
parent ee04aead1e
commit fad74b701b
476 changed files with 1270 additions and 46 deletions

View File

@@ -204,9 +204,6 @@ def which_attn_to_use(
if selected_backend != _Backend.PALLAS:
logger.info("Cannot use %s backend on TPU.", selected_backend)
return _Backend.PALLAS
if selected_backend == _Backend.FLASH_ATTN:
print("selected_backend == _Backend.FLASH_ATTN")
if is_hip():
# AMD GPUs.