Unverified Commit d4bfc23e authored by Harry Mellor's avatar Harry Mellor Committed by GitHub
Browse files

Fix Transformers backend compatibility check (#15290)


Signed-off-by: default avatarHarry Mellor <19981378+hmellor@users.noreply.github.com>
parent 9a2160fa
......@@ -36,10 +36,7 @@ def is_transformers_impl_compatible(
mod = module or getattr(transformers, arch, None)
if mod is None:
return False
if hasattr(mod, "supports_backend"):
return mod.is_backend_compatible()
else:
return mod._supports_flex_attn
return mod.is_backend_compatible()
def resolve_transformers_fallback(model_config: ModelConfig,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment