Skip to content

Comments

Add option for QKV clipping#489

Merged
epwalsh merged 2 commits intomainfrom
epwalsh/qkv-clip
Mar 7, 2024
Merged

Add option for QKV clipping#489
epwalsh merged 2 commits intomainfrom
epwalsh/qkv-clip

Conversation

@epwalsh
Copy link
Member

@epwalsh epwalsh commented Mar 7, 2024

Adds a config field for enabling QKV clipping: clip_qkv: Optional[float].

@epwalsh epwalsh requested review from AkshitaB and dirkgr March 7, 2024 00:09
Copy link
Contributor

@AkshitaB AkshitaB left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks fine. Should we add a check for the clipping value to be positive, since we negate it for getting the min value?

@epwalsh epwalsh merged commit c499632 into main Mar 7, 2024
@epwalsh epwalsh deleted the epwalsh/qkv-clip branch March 7, 2024 00:25
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants