Improve comments#328

Draft

kfeng123 wants to merge 1 commit intoaqlaboratory:mainfrom

kfeng123:typo_fix

Commits on Jun 16, 2023

Improve comments, especially for TriangleAttention: c_hidden is the per-head hidden dimension, not overall hidden dimension.
kfeng123
committed