Skip to content

Migrate SelfAttention and CrossAttention off of ModuleSpec - #2454

Closed
nschank wants to merge 8 commits into
NVIDIA:mainfrom
nschank:typespec1
Closed

Migrate SelfAttention and CrossAttention off of ModuleSpec#2454
nschank wants to merge 8 commits into
NVIDIA:mainfrom
nschank:typespec1

Add Protocol-based typing for SelfAttention.layernorm

9256eb5
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs