Hugging Face's logo

Spaces:
yanyichu
/
UTR_LM
Running

App Files Community
2
yanyichu commited on
Commit
b4a1072
•
1 Parent(s): e2615ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.pkl'
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