kanshan/diffusers
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

changed to posix unet (#6719)

Browse Source 
changed to posix

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
This commit is contained in:
gzguevara
2024-01-27 12:01:52 +01:00
committed by GitHub
parent c101066227
commit 8581d9bce4
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/diffusers/loaders/unet.py
+4 -2
View File
@@ -16,6 +16,7 @@ import os
from collections import defaultdict from collections import defaultdict
from contextlib import nullcontext from contextlib import nullcontext
from functools import partial from functools import partial
from pathlib import Path
from typing import Callable, Dict, List, Optional, Union from typing import Callable, Dict, List, Optional, Union
import safetensors import safetensors
@@ -503,8 +504,9 @@ class UNet2DConditionLoadersMixin:
weight_name = CUSTOM_DIFFUSION_WEIGHT_NAME if is_custom_diffusion else LORA_WEIGHT_NAME weight_name = CUSTOM_DIFFUSION_WEIGHT_NAME if is_custom_diffusion else LORA_WEIGHT_NAME
# Save the model # Save the model
save_function(state_dict, os.path.join(save_directory, weight_name)) save_path = Path(save_directory, weight_name).as_posix()
logger.info(f"Model weights saved in {os.path.join(save_directory, weight_name)}") 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): def fuse_lora(self, lora_scale=1.0, safe_fusing=False, adapter_names=None):
self.lora_scale = lora_scale self.lora_scale = lora_scale