Hugging Face's logo

Spaces:
damerajee
/
mingru-stories
Sleeping

App Files Community
damerajee commited on
Commit
6941429
·
verified ·
1 Parent(s): 85b3986

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from mingru_lm import MinGRU_LM
4
 
5
 
6
  model = MinGRU_LM(dim=512, num_tokens=256, num_layers=6)
7
- pt_model = "model/best_model.pt"
8
  checkpoint = torch.load(pt_model)
9
  model.load_state_dict(checkpoint['model_state_dict'])
10
 
 
4
 
5
 
6
  model = MinGRU_LM(dim=512, num_tokens=256, num_layers=6)
7
+ pt_model = "best_model.pt"
8
  checkpoint = torch.load(pt_model)
9
  model.load_state_dict(checkpoint['model_state_dict'])
10