Update modeling.py
Browse files- modeling.py +1 -1
modeling.py
CHANGED
|
@@ -114,7 +114,7 @@ class MARModel(PreTrainedModel):
|
|
| 114 |
def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):
|
| 115 |
config = MARConfig.from_pretrained(pretrained_model_name_or_path, *model_args, **kwargs)
|
| 116 |
model = cls(config)
|
| 117 |
-
state_dict = torch.load('
|
| 118 |
model.model.load_state_dict(state_dict)
|
| 119 |
return model
|
| 120 |
|
|
|
|
| 114 |
def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):
|
| 115 |
config = MARConfig.from_pretrained(pretrained_model_name_or_path, *model_args, **kwargs)
|
| 116 |
model = cls(config)
|
| 117 |
+
state_dict = torch.load('checkpoint-last.safetensors')
|
| 118 |
model.model.load_state_dict(state_dict)
|
| 119 |
return model
|
| 120 |
|