Skip to content

Commit 01e0f0e

Browse files
hmellorlk-chen
authored andcommitted
Fix Transformers backend compatibility check (vllm-project#15290)
Signed-off-by: Harry Mellor <[email protected]>
1 parent 1e03b82 commit 01e0f0e

File tree

1 file changed

+1
-4
lines changed
  • vllm/model_executor/model_loader

1 file changed

+1
-4
lines changed

vllm/model_executor/model_loader/utils.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,10 +36,7 @@ def is_transformers_impl_compatible(
3636
mod = module or getattr(transformers, arch, None)
3737
if mod is None:
3838
return False
39-
if hasattr(mod, "supports_backend"):
40-
return mod.is_backend_compatible()
41-
else:
42-
return mod._supports_flex_attn
39+
return mod.is_backend_compatible()
4340

4441

4542
def resolve_transformers_fallback(model_config: ModelConfig,

0 commit comments

Comments
 (0)