(
self,
save_directory: Union[str, os.PathLike],
safe_serialization: bool = True,
**kwargs,
)
| 320 | return module_obj |
| 321 | |
| 322 | def save_pretrained( |
| 323 | self, |
| 324 | save_directory: Union[str, os.PathLike], |
| 325 | safe_serialization: bool = True, |
| 326 | **kwargs, |
| 327 | ): |
| 328 | push_to_hub = kwargs.pop('push_to_hub', False) |
| 329 | if push_to_hub: |
| 330 | from modelscope.hub.push_to_hub import push_to_hub |
| 331 | from modelscope.hub.api import HubApi |
| 332 | from modelscope.hub.repository import Repository |
| 333 | |
| 334 | token = kwargs.get('token') |
| 335 | commit_message = kwargs.pop('commit_message', None) |
| 336 | repo_name = kwargs.pop( |
| 337 | 'repo_id', |
| 338 | save_directory.split(os.path.sep)[-1]) |
| 339 | |
| 340 | api = HubApi() |
| 341 | api.login(token) |
| 342 | api.create_repo(repo_name) |
| 343 | # clone the repo |
| 344 | Repository(save_directory, repo_name) |
| 345 | |
| 346 | super().save_pretrained( |
| 347 | save_directory=save_directory, |
| 348 | safe_serialization=safe_serialization, |
| 349 | push_to_hub=False, |
| 350 | **kwargs) |
| 351 | |
| 352 | # Class members may be unpatched, so push_to_hub is done separately here |
| 353 | if push_to_hub: |
| 354 | push_to_hub( |
| 355 | repo_name=repo_name, |
| 356 | output_dir=save_directory, |
| 357 | commit_message=commit_message, |
| 358 | token=token) |
| 359 | |
| 360 | if not hasattr(module_class, 'from_pretrained'): |
| 361 | del ClassWrapper.from_pretrained |
nothing calls this directly
no test coverage detected