From 2dc7fc2a6eb0ce4a4a9d434bcadd6bc6179b55db Mon Sep 17 00:00:00 2001 From: charchit7 Date: Tue, 16 Jan 2024 11:40:28 +0530 Subject: [PATCH 1/2] posix lora --- src/diffusers/loaders/lora.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/diffusers/loaders/lora.py b/src/diffusers/loaders/lora.py index 7df91a05b593..fec6d06433f3 100644 --- a/src/diffusers/loaders/lora.py +++ b/src/diffusers/loaders/lora.py @@ -960,8 +960,9 @@ def save_function(weights, filename): else: weight_name = LORA_WEIGHT_NAME - 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 unload_lora_weights(self): """ From 991dcbd841f424cb2b550747e3a643c7b1f5b4dd Mon Sep 17 00:00:00 2001 From: charchit7 Date: Tue, 16 Jan 2024 11:43:45 +0530 Subject: [PATCH 2/2] changes and style fix --- src/diffusers/loaders/lora.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/diffusers/loaders/lora.py b/src/diffusers/loaders/lora.py index fec6d06433f3..922c98b98bf4 100644 --- a/src/diffusers/loaders/lora.py +++ b/src/diffusers/loaders/lora.py @@ -14,6 +14,7 @@ import inspect import os from contextlib import nullcontext +from pathlib import Path from typing import Callable, Dict, List, Optional, Union import safetensors