1580158
/
sd-webui-supermerger
mirror of https://github.com/hako-mikan/sd-webui-supermerger.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

#354,#365

ver18
hako-mikan 2024-04-04 20:27:55 +09:00 committed by GitHub
parent 3c90a58d94
commit 9b7e3e003c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/kohyas/sdxl_model_util.py
View File

@ -4,7 +4,7 @@ from accelerate.utils.modeling import set_module_tensor_to_device
from safetensors.torch import load_file, save_file
from transformers import CLIPTextModel, CLIPTextConfig, CLIPTextModelWithProjection, CLIPTokenizer
from typing import List
from diffusers import AutoencoderKL, EulerDiscreteScheduler, UNet2DConditionModel
from diffusers import AutoencoderKL, UNet2DConditionModel
from scripts.kohyas import model_util, sdxl_original_unet
@ -537,7 +537,7 @@ def save_diffusers_checkpoint(
if pretrained_model_name_or_path is None:
pretrained_model_name_or_path = DIFFUSERS_REF_MODEL_ID_SDXL
scheduler = EulerDiscreteScheduler.from_pretrained(pretrained_model_name_or_path, subfolder="scheduler")
#scheduler = EulerDiscreteScheduler.from_pretrained(pretrained_model_name_or_path, subfolder="scheduler")
tokenizer1 = CLIPTokenizer.from_pretrained(pretrained_model_name_or_path, subfolder="tokenizer")
tokenizer2 = CLIPTokenizer.from_pretrained(pretrained_model_name_or_path, subfolder="tokenizer_2")
if vae is None: