"wrappers/python/vscode:/vscode.git/clone" did not exist on "911e27daf320580dc9ab89954b2285a5778fa59f"
Unverified Commit a2e9ebe9 authored by Fanli Lin's avatar Fanli Lin Committed by GitHub
Browse files

[BugFix] Fix flash_attn import in `siglip2navit.py` (#29082)


Signed-off-by: default avatarFanli Lin <fanli.lin@intel.com>
parent 93c8672c
......@@ -191,7 +191,7 @@ def apply_rotary_pos_emb(
cos = cos.chunk(2, dim=-1)[0].contiguous()
sin = sin.chunk(2, dim=-1)[0].contiguous()
if is_flash_attn_backend and not current_platform.is_xpu():
from flash_attn.layers.rotary import apply_rotary_emb
from vllm.vllm_flash_attn.layers.rotary import apply_rotary_emb
apply_rotary_emb_func = apply_rotary_emb
else:
......
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