Update qwenimage/qwen_fa3_processor.py

qwenimage/qwen_fa3_processor.py

@@ -1,3 +1,7 @@
+"""
+Paired with a good language model. Thanks!
+"""
+
 import torch
 from typing import Optional, Tuple
 from diffusers.models.transformers.transformer_qwenimage import apply_rotary_emb_qwen
@@ -18,14 +22,11 @@ def _ensure_fa3_available():
             "Tried `get_kernel('kernels-community/vllm-flash-attn3')` and failed with:\n"
             f"{_kernels_err}"
         )
-            
 
 @torch.library.custom_op("flash::flash_attn_func", mutates_args=())
 def flash_attn_func(
     q: torch.Tensor, k: torch.Tensor, v: torch.Tensor, causal: bool = False
 ) -> torch.Tensor:
-    from flash_attn.flash_attn_interface import flash_attn_interface_func
-    
     outputs, lse = _flash_attn_func(q, k, v, causal=causal)
     return outputs