|
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