Add argument types to be able to use torch JIT#54

Open

AdrianEddy wants to merge 3 commits intoautonomousvision:masterfrom AdrianEddy:master

+465-410

Commits on Apr 1, 2024

Commits on May 10, 2024

Fix generate_shift_window_attn_mask_1d
AdrianEddy
committed