Remove custom pre attention scaling and use computed value instead.
This commit is contained in:
parent
a89427908d
commit
51f0bd50a1
4 changed files with 2 additions and 13 deletions
|
@ -52,7 +52,6 @@ class Keys:
|
|||
DECODER_START_TOKEN_ID = "{arch}.decoder_start_token_id"
|
||||
ATTN_LOGIT_SOFTCAPPING = "{arch}.attn_logit_softcapping"
|
||||
FINAL_LOGIT_SOFTCAPPING = "{arch}.final_logit_softcapping"
|
||||
QUERY_PRE_ATTN_SCALAR = "{arch}.query_pre_attn_scalar"
|
||||
|
||||
class Attention:
|
||||
HEAD_COUNT = "{arch}.attention.head_count"
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue