Fix/janus pro flash attention#152

Open

vadseshu wants to merge 2 commits intoROCm:developfrom

vadseshu:fix/janus_pro_flash_attention

Commits on Apr 16, 2026

Added gfx950a to pyt_mochi_inference.ubuntu dockerfile to enable fa
vadseshu1
committed

Commits on Apr 22, 2026

Changed FA steps and fixing runtime failures wrt transformers-5.x
vadseshu
committed