Skip to content

Commit 2389efd

Browse files
Update phishing_email_detection_gpt2.py
Fix name of arg.
1 parent e3ad67d commit 2389efd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

phishing_email_detection_gpt2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -264,7 +264,7 @@ def call(self, x):
264264

265265
position_embedding = RotaryPositionEmbedding(
266266
sequence_length=max_seq_length,
267-
sequence_length=EMBEDDING_DIM
267+
d_model=EMBEDDING_DIM,
268268
# initializer="uniform",
269269
)(embedded)
270270

0 commit comments

Comments
 (0)