[diffusion] CI: improve diffusion CI (#13562)
Co-authored-by: Adarsh Shirawalmath <114558126+adarshxs@users.noreply.github.com>
This commit is contained in:
co-authored by
Adarsh Shirawalmath
parent
af6bcadcf7
commit
127d59cd2c
@@ -69,8 +69,8 @@ class RocmPlatform(Platform):
|
||||
dtype: torch.dtype,
|
||||
) -> str:
|
||||
logger.info(
|
||||
"Trying SGL_DIFFUSION_ATTENTION_BACKEND=%s",
|
||||
envs.SGL_DIFFUSION_ATTENTION_BACKEND,
|
||||
"Trying SGLANG_DIFFUSION_ATTENTION_BACKEND=%s",
|
||||
envs.SGLANG_DIFFUSION_ATTENTION_BACKEND,
|
||||
)
|
||||
|
||||
if selected_backend == AttentionBackendEnum.TORCH_SDPA:
|
||||
|
||||
Reference in New Issue
Block a user