Lise Le Boudec
Results
1
comments of
Lise Le Boudec
Hi, +1, I would be very interested by this feature also! If it can help, changing line 499 in flash-linear-attention/fla/ops/common/chunk_o.py from ` o = torch.empty_like(v)` to ` o = torch.empty_like(q)`...