Skip to content

[PyTorch/Jax] Fix attention mask definition, and sliding window for decoder - #818

Merged
ptrendx merged 29 commits into
NVIDIA:mainfrom
cyanguwa:fix_attn_masks
May 17, 2024
Merged

[PyTorch/Jax] Fix attention mask definition, and sliding window for decoder#818
ptrendx merged 29 commits into
NVIDIA:mainfrom
cyanguwa:fix_attn_masks

expand list to tensors for save_for_backwards

f07b0bf
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs