Neu256 commited on
Commit
8523487
·
verified ·
1 Parent(s): 2fa4adb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -10,7 +10,6 @@ from utils import (
10
  NUM_EMBED,
11
  NUM_HEAD,
12
  NUM_LAYER,
13
- BLOCK_SIZE,
14
  encode,
15
  decode
16
  )
@@ -22,7 +21,6 @@ vocab_size = tokenizer.vocab_size
22
  model = Transformer(
23
  vocab_size=vocab_size,
24
  num_embed=NUM_EMBED,
25
- block_size=BLOCK_SIZE,
26
  num_heads=NUM_HEAD,
27
  num_layers=NUM_LAYER,
28
  dropout=DROPOUT
 
10
  NUM_EMBED,
11
  NUM_HEAD,
12
  NUM_LAYER,
 
13
  encode,
14
  decode
15
  )
 
21
  model = Transformer(
22
  vocab_size=vocab_size,
23
  num_embed=NUM_EMBED,
 
24
  num_heads=NUM_HEAD,
25
  num_layers=NUM_LAYER,
26
  dropout=DROPOUT