Spaces:
Running
on
Zero
Running
on
Zero
Update qwenimage/qwen_fa3_processor.py
Browse files
qwenimage/qwen_fa3_processor.py
CHANGED
|
@@ -1,4 +1,6 @@
|
|
| 1 |
import torch
|
|
|
|
|
|
|
| 2 |
|
| 3 |
@torch.library.custom_op("flash::flash_attn_func", mutates_args=())
|
| 4 |
def flash_attn_func(
|
|
|
|
| 1 |
import torch
|
| 2 |
+
from typing import Optional, Tuple
|
| 3 |
+
from diffusers.models.transformers.transformer_qwenimage import apply_rotary_emb_qwen
|
| 4 |
|
| 5 |
@torch.library.custom_op("flash::flash_attn_func", mutates_args=())
|
| 6 |
def flash_attn_func(
|