From 9a8dea729076dc95d8b969e1b6fad439721b2717 Mon Sep 17 00:00:00 2001 From: Limitex <76650151+Limitex@users.noreply.github.com> Date: Tue, 2 Jan 2024 15:18:24 +0900 Subject: [PATCH] Add referrer comment --- utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils.py b/utils.py index c387152..d4a8402 100644 --- a/utils.py +++ b/utils.py @@ -70,7 +70,7 @@ def token_auto_concat_embeds(pipe, positive, negative): negative_prompt_embeds = torch.cat(negative_concat_embeds, dim=1) return positive_prompt_embeds, negative_prompt_embeds - +# Reference from : https://github.com/huggingface/diffusers/blob/main/scripts/convert_vae_pt_to_diffusers.py def custom_convert_ldm_vae_checkpoint(checkpoint, config): vae_state_dict = checkpoint @@ -171,7 +171,7 @@ def custom_convert_ldm_vae_checkpoint(checkpoint, config): conv_attn_to_linear(new_checkpoint) return new_checkpoint - +# Reference from : https://github.com/huggingface/diffusers/blob/main/scripts/convert_vae_pt_to_diffusers.py def vae_pt_to_vae_diffuser( checkpoint_path: str, output_path: str,