fix: MaskedCausalVisionTransformer build on all supported timm versions - #2043

Merged

gabrielfruet merged 2 commits into

lightly-ai:masterfrom

lorinczszabolcs:fix-masked-causal-vit-attention-kwargs

Aug 21, 2026

Commits on Aug 21, 2026

fix: forward only attention kwargs in MaskedCausalBlock
lorinczszabolcs
committed
e5ab291
fix: set global_pool for the AIM masked causal vision transformer
lorinczszabolcs
committed
9ee3d4e