Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ seed = 1337
|
|
16 |
torch.manual_seed(seed)
|
17 |
|
18 |
global modelfile, layers, heads, embed_dim, batch_toks, inp_len, device
|
19 |
-
modelfile = 'model.
|
20 |
|
21 |
layers = 6
|
22 |
heads = 16
|
|
|
16 |
torch.manual_seed(seed)
|
17 |
|
18 |
global modelfile, layers, heads, embed_dim, batch_toks, inp_len, device
|
19 |
+
modelfile = 'model.pt'
|
20 |
|
21 |
layers = 6
|
22 |
heads = 16
|