We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 151b818 commit c9f6af1Copy full SHA for c9f6af1
discojs/discojs-core/src/models/gpt/model.ts
@@ -116,7 +116,7 @@ serialization.registerClass(CausalSelfAttentionBase)
116
117
type CausalSelfAttentionConfig =
118
ConstructorParameters<typeof tf.layers.Layer>[0]
119
- & Record<'blockSize' | 'nHead' | 'nEmbd' | 'dropout' | 'mask', number>
+ & Record<'blockSize' | 'nHead' | 'nEmbd' | 'dropout', number>
120
& { bias: boolean }
121
122
class CausalSelfAttention extends layers.Layer {
0 commit comments