Update to new version of base image
This commit is contained in:
@@ -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.
|
||||
|
||||
Reference in New Issue
Block a user