From 8581d9bce43ac2747199f848a9ac861352495d10 Mon Sep 17 00:00:00 2001 From: gzguevara <55751398+gzguevara@users.noreply.github.com> Date: Sat, 27 Jan 2024 12:01:52 +0100 Subject: [PATCH] changed to posix unet (#6719) changed to posix Co-authored-by: Sayak Paul --- src/diffusers/loaders/unet.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/diffusers/loaders/unet.py b/src/diffusers/loaders/unet.py index 11a32a92ae..d7c145e430 100644 --- a/src/diffusers/loaders/unet.py +++ b/src/diffusers/loaders/unet.py @@ -16,6 +16,7 @@ import os from collections import defaultdict from contextlib import nullcontext from functools import partial +from pathlib import Path from typing import Callable, Dict, List, Optional, Union import safetensors @@ -503,8 +504,9 @@ class UNet2DConditionLoadersMixin: weight_name = CUSTOM_DIFFUSION_WEIGHT_NAME if is_custom_diffusion else LORA_WEIGHT_NAME # Save the model - save_function(state_dict, os.path.join(save_directory, weight_name)) - logger.info(f"Model weights saved in {os.path.join(save_directory, weight_name)}") + save_path = Path(save_directory, weight_name).as_posix() + save_function(state_dict, save_path) + logger.info(f"Model weights saved in {save_path}") def fuse_lora(self, lora_scale=1.0, safe_fusing=False, adapter_names=None): self.lora_scale = lora_scale