Skip to content

Commit 0e5e23e

Browse files
authored
Restore the hub_mixin.py file
1 parent e40022b commit 0e5e23e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

segmentation_models_pytorch/base/hub_mixin.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ def save_pretrained(
120120
finally:
121121
# delete the additional attributes
122122
self._del_attrs(["save_directory", "metrics", "dataset"])
123-
# self._hub_mixin_config.pop("_model_class")
123+
self._hub_mixin_config.pop("_model_class")
124124

125125
return result
126126

0 commit comments

Comments
 (0)