ignore keys for now
Browse files- modeling.py +1 -1
modeling.py
CHANGED
|
@@ -135,7 +135,7 @@ class MARModel(PreTrainedModel):
|
|
| 135 |
|
| 136 |
state_dict = torch.load(checkpoint_path, map_location=device)["model_ema"]
|
| 137 |
|
| 138 |
-
model_architecture.load_state_dict(state_dict)
|
| 139 |
|
| 140 |
model_architecture.eval()
|
| 141 |
|
|
|
|
| 135 |
|
| 136 |
state_dict = torch.load(checkpoint_path, map_location=device)["model_ema"]
|
| 137 |
|
| 138 |
+
model_architecture.load_state_dict(state_dict, strict=False)
|
| 139 |
|
| 140 |
model_architecture.eval()
|
| 141 |
|