recursionaut commited on
Commit
2c51125
1 Parent(s): 3eb4127

fix: minor change

Browse files
Files changed (1) hide show
  1. huggingface_mae.py +1 -1
huggingface_mae.py CHANGED
@@ -288,6 +288,6 @@ class MAEModel(PreTrainedModel):
288
  modelpath = f"{pretrained_model_name_or_path}/{filename}"
289
  config = MAEConfig.from_pretrained(pretrained_model_name_or_path, **kwargs)
290
  state_dict = torch.load(modelpath, map_location="cpu")
291
- model = cls(config, *model_args, **kwargs)
292
  model.load_state_dict(state_dict["state_dict"])
293
  return model
 
288
  modelpath = f"{pretrained_model_name_or_path}/{filename}"
289
  config = MAEConfig.from_pretrained(pretrained_model_name_or_path, **kwargs)
290
  state_dict = torch.load(modelpath, map_location="cpu")
291
+ model = cls(config)
292
  model.load_state_dict(state_dict["state_dict"])
293
  return model