From 6bbb4d426efb97607dd2d587412750da10164d17 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Sun, 29 Jan 2023 20:43:58 +0900 Subject: [PATCH] Fix unet config in Diffusers (sample_size=64) --- library/model_util.py | 2 +- tools/convert_diffusers20_original_sd.py | 4 ---- 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/library/model_util.py b/library/model_util.py index 6a1e656a..778a4c27 100644 --- a/library/model_util.py +++ b/library/model_util.py @@ -16,7 +16,7 @@ BETA_END = 0.0120 UNET_PARAMS_MODEL_CHANNELS = 320 UNET_PARAMS_CHANNEL_MULT = [1, 2, 4, 4] UNET_PARAMS_ATTENTION_RESOLUTIONS = [4, 2, 1] -UNET_PARAMS_IMAGE_SIZE = 32 # unused +UNET_PARAMS_IMAGE_SIZE = 64 # fixed from old invalid value `32` UNET_PARAMS_IN_CHANNELS = 4 UNET_PARAMS_OUT_CHANNELS = 4 UNET_PARAMS_NUM_RES_BLOCKS = 2 diff --git a/tools/convert_diffusers20_original_sd.py b/tools/convert_diffusers20_original_sd.py index a3cd03fe..6c142848 100644 --- a/tools/convert_diffusers20_original_sd.py +++ b/tools/convert_diffusers20_original_sd.py @@ -1,8 +1,4 @@ # convert Diffusers v1.x/v2.0 model to original Stable Diffusion -# v1: initial version -# v2: support safetensors -# v3: fix to support another format -# v4: support safetensors in Diffusers import argparse import os