|
20 | 20 | import re
|
21 | 21 | from collections import OrderedDict
|
22 | 22 | from functools import partial
|
| 23 | +from pathlib import Path |
23 | 24 | from typing import Any, Callable, List, Optional, Tuple, Union
|
24 | 25 |
|
25 | 26 | import safetensors
|
@@ -367,18 +368,18 @@ def save_pretrained(
|
367 | 368 | # Save the model
|
368 | 369 | if safe_serialization:
|
369 | 370 | safetensors.torch.save_file(
|
370 |
| - state_dict, os.path.join(save_directory, weights_name), metadata={"format": "pt"} |
| 371 | + state_dict, Path(save_directory, weights_name).as_posix(), metadata={"format": "pt"} |
371 | 372 | )
|
372 | 373 | else:
|
373 |
| - torch.save(state_dict, os.path.join(save_directory, weights_name)) |
| 374 | + torch.save(state_dict, Path(save_directory, weights_name).as_posix()) |
374 | 375 |
|
375 |
| - logger.info(f"Model weights saved in {os.path.join(save_directory, weights_name)}") |
| 376 | + logger.info(f"Model weights saved in {Path(save_directory, weights_name).as_posix()}") |
376 | 377 |
|
377 | 378 | if push_to_hub:
|
378 | 379 | # Create a new empty model card and eventually tag it
|
379 | 380 | model_card = load_or_create_model_card(repo_id, token=token)
|
380 | 381 | model_card = populate_model_card(model_card)
|
381 |
| - model_card.save(os.path.join(save_directory, "README.md")) |
| 382 | + model_card.save(Path(save_directory, "README.md").as_posix()) |
382 | 383 |
|
383 | 384 | self._upload_folder(
|
384 | 385 | save_directory,
|
|
0 commit comments