0 0 1 Python

Set member variable to Attention module (#6376)

- Set and use self.n_heads from args.heads
- Use self.dim from args.dim

Co-authored-by: Chun-I Tsai <chunit@hu-chunit-sha.qualcomm.com>

Chun-I Tsai committed 1y ago

904a2e51484329eff50236df7afc9e38970182cc

Parent: f044e91

Committed by GitHub <noreply@github.com> on 10/25/2024, 10:29:43 PM