Spaces:
Runtime error
Runtime error
Commit
·
215cb38
1
Parent(s):
00d213f
Update utils.py
Browse files
utils.py
CHANGED
@@ -7,9 +7,9 @@ chars = ['\n', ' ', '!', '$', '&', "'", ',', '-', '.', '3', ':', ';', '?', 'A',
|
|
7 |
|
8 |
vocab_size = len(chars)
|
9 |
block_size = 32
|
10 |
-
n_embd =
|
11 |
n_head = 4
|
12 |
-
n_layer =
|
13 |
dropout = 0.1
|
14 |
|
15 |
class Head(nn.Module):
|
|
|
7 |
|
8 |
vocab_size = len(chars)
|
9 |
block_size = 32
|
10 |
+
n_embd = 128
|
11 |
n_head = 4
|
12 |
+
n_layer = 8
|
13 |
dropout = 0.1
|
14 |
|
15 |
class Head(nn.Module):
|