diff --git a/src/diffusers/models/modeling_utils.py b/src/diffusers/models/modeling_utils.py index cea76c53d945..ccc3536079c6 100644 --- a/src/diffusers/models/modeling_utils.py +++ b/src/diffusers/models/modeling_utils.py @@ -20,6 +20,7 @@ import re from collections import OrderedDict from functools import partial +from pathlib import Path from typing import Any, Callable, List, Optional, Tuple, Union import safetensors @@ -369,18 +370,18 @@ def save_pretrained( # Save the model if safe_serialization: safetensors.torch.save_file( - state_dict, os.path.join(save_directory, weights_name), metadata={"format": "pt"} + state_dict, Path(save_directory, weights_name).as_posix(), metadata={"format": "pt"} ) else: - torch.save(state_dict, os.path.join(save_directory, weights_name)) + torch.save(state_dict, Path(save_directory, weights_name).as_posix()) - logger.info(f"Model weights saved in {os.path.join(save_directory, weights_name)}") + logger.info(f"Model weights saved in {Path(save_directory, weights_name).as_posix()}") if push_to_hub: # Create a new empty model card and eventually tag it model_card = load_or_create_model_card(repo_id, token=token) model_card = populate_model_card(model_card) - model_card.save(os.path.join(save_directory, "README.md")) + model_card.save(Path(save_directory, "README.md").as_posix()) self._upload_folder( save_directory,