Skip to content

Commit c9f6af1

Browse files
committedMar 13, 2024
fixup! discojs-core/models: add gpt
1 parent 151b818 commit c9f6af1

File tree

1 file changed

+1
-1
lines changed
  • discojs/discojs-core/src/models/gpt

1 file changed

+1
-1
lines changed
 

Diff for: ‎discojs/discojs-core/src/models/gpt/model.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ serialization.registerClass(CausalSelfAttentionBase)
116116

117117
type CausalSelfAttentionConfig =
118118
ConstructorParameters<typeof tf.layers.Layer>[0]
119-
& Record<'blockSize' | 'nHead' | 'nEmbd' | 'dropout' | 'mask', number>
119+
& Record<'blockSize' | 'nHead' | 'nEmbd' | 'dropout', number>
120120
& { bias: boolean }
121121

122122
class CausalSelfAttention extends layers.Layer {

0 commit comments

Comments
 (0)