Fix typo
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ n_head = 16
|
|
26 |
n_layers = 8
|
27 |
|
28 |
#device = "cuda"
|
29 |
-
|
30 |
|
31 |
def encode_pad(s):
|
32 |
if len(s) >= block_size:
|
|
|
26 |
n_layers = 8
|
27 |
|
28 |
#device = "cuda"
|
29 |
+
device = "cpu"
|
30 |
|
31 |
def encode_pad(s):
|
32 |
if len(s) >= block_size:
|