Commit f40b6ed

authored

Update hub_mixin.py (#909)

Fix the error caused by pop operation on non-existing key in dict.

1 parent c68fc94 commit f40b6edCopy full SHA for f40b6ed

File tree

-1

lines changed

-1

lines changed

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -120,7 +120,7 @@ def save_pretrained(`
`120`	`120`	`finally:`
`121`	`121`	`# delete the additional attributes`
`122`	`122`	`self._del_attrs(["save_directory", "metrics", "dataset"])`
`123`		`- self._hub_mixin_config.pop("_model_diff-ddc564a4f580a1a0e674d7b0312555d2c3a29d2a025e53de685c156a0bf7f745-123-123-0" data-selected="false" role="gridcell" tabindex="-1">`	`123`	`+ self._hub_mixin_config.pop("_model_class", None)`
`124`	`124`
`125`	`125`	`return result`
`126`	`126`

Comments

(0)