jadechoghari commited on
Commit
b4c4545
·
verified ·
1 Parent(s): 5309bad

Update modeling.py

Browse files
Files changed (1) hide show
  1. 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('./checkpoint-last.safetensors')
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