Skip to content

Commit a2e9ebe

Browse files
authored
[BugFix] Fix flash_attn import in siglip2navit.py (#29082)
Signed-off-by: Fanli Lin <[email protected]>
1 parent 93c8672 commit a2e9ebe

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/model_executor/models/siglip2navit.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -191,7 +191,7 @@ def apply_rotary_pos_emb(
191191
cos = cos.chunk(2, dim=-1)[0].contiguous()
192192
sin = sin.chunk(2, dim=-1)[0].contiguous()
193193
if is_flash_attn_backend and not current_platform.is_xpu():
194-
from flash_attn.layers.rotary import apply_rotary_emb
194+
from vllm.vllm_flash_attn.layers.rotary import apply_rotary_emb
195195

196196
apply_rotary_emb_func = apply_rotary_emb
197197
else:

0 commit comments

Comments
 (0)