Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ from mingru_lm import MinGRU_LM
|
|
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,map_location=torch.device('cpu')
|
9 |
model.load_state_dict(checkpoint['model_state_dict'])
|
10 |
|
11 |
# Move model to GPU if available
|
|
|
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,map_location=torch.device('cpu'))
|
9 |
model.load_state_dict(checkpoint['model_state_dict'])
|
10 |
|
11 |
# Move model to GPU if available
|