Skip to content

Commit

Permalink
fixed loading from hub
Browse files Browse the repository at this point in the history
  • Loading branch information
Butanium committed Nov 22, 2024
1 parent 5256d5b commit b487641
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dictionary_learning/dictionary.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ def from_pretrained(
device: Device to load the model to
**kwargs: Additional arguments passed to loading function
"""
model = PyTorchModelHubMixin.from_pretrained(cls, path, **kwargs)
model = super(Dictionary, cls).from_pretrained(path, **kwargs)
if device is not None:
model.to(device)
if dtype is not None:
Expand Down

0 comments on commit b487641

Please sign in to comment.