Spaces:
Running
Running
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
|