text-generation-inference
feat: support force downcast after FastRMSNorm multiply for Gemma
#1658
Merged

feat: support force downcast after FastRMSNorm multiply for Gemma #1658

Narsil merged 3 commits into main from fix-gemma-bugs
drbh
drbh feat: support force downcast after FastRMSNorm multiply
b307fce6
Narsil
Narsil commented on 2024-03-20
drbh feat: prefer gemma specific rms
5b076dfc
drbh fix: simplify syntax
704d4ddf
Narsil
Narsil approved these changes on 2024-03-21
Narsil Narsil merged 6f15ac60 into main 1 year ago
Narsil Narsil deleted the fix-gemma-bugs branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone