We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c68fc94 commit f40b6edCopy full SHA for f40b6ed
segmentation_models_pytorch/base/hub_mixin.py
@@ -120,7 +120,7 @@ def save_pretrained(
120
finally:
121
# delete the additional attributes
122
self._del_attrs(["save_directory", "metrics", "dataset"])
123
- self._hub_mixin_config.pop("_model_class")
+ self._hub_mixin_config.pop("_model_class", None)
124
125
return result
126
0 commit comments