Fixes flash_attn + cascade attention_code to decoder Transformer bloc…#71
Merged
LBerth merged 5 commits intometeofrance:mainfrom colon3ltocard:unetrpp_flash_attn_decoder_headsOct 10, 2024
+125-18
Commits
Commits on Oct 10, 2024
- committed
- committed
Merge branch 'main' of https://github.com/colon3ltocard/py4cast into unetrpp_flash_attn_decoder_heads
committed- committed
- committed
