fix figure test bug.

python/sglang/srt/layers/attention/vision.py

@@ -29,6 +29,9 @@ _is_hip = is_hip()
 if _is_cuda:
     from sgl_kernel.flash_attn import flash_attn_varlen_func
 
+if _is_hip:
+    from sglang.srt.layers.attention.flashattention_interface import flash_attn_varlen_func
+
 if _is_npu:
     import torch_npu
 
@@ -299,8 +302,6 @@ class VisionFlash3Attention(nn.Module):
     ):
         # if not _is_cuda:
         #     raise Exception("VisionFlash3Attention is only available for cuda")
-        if _is_hip:
-            from sglang.srt.layers.attention.flashattention_interface import flash_attn_varlen_func
         super().__init__()
 
     def forward(