Skip to content

feat: Add TP to embed_tokens and lm_head for Gemma models#879

Merged
terrykong merged 1 commit into
mainfrom
ruit/add_tp_for_gemma
Aug 14, 2025
Merged

feat: Add TP to embed_tokens and lm_head for Gemma models#879
terrykong merged 1 commit into
mainfrom
ruit/add_tp_for_gemma

Add TP to embed_tokens and lm_head for Gemma models

4ebac48
Select commit
Loading
Failed to load commit list.
Codecov / codecov/patch succeeded Aug 12, 2025 in 1s

Coverage not affected when comparing 223bfa8...4ebac48

View this Pull Request on Codecov

Coverage not affected when comparing 223bfa8...4ebac48

View more details on Codecov