From 1ab2f3f4d6cb39b5a8cea65184bbb05fe3f8b56d Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Wed, 24 Jul 2024 20:49:52 +0300 Subject: [PATCH] fix Comfy model loader --- nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index 4f3c832..128ffec 100644 --- a/nodes.py +++ b/nodes.py @@ -746,7 +746,7 @@ class ADMD_ComfyModelLoader: original_config = OmegaConf.load(os.path.join(script_directory, f"configs/v1-inference.yaml")) ad_unet_config = OmegaConf.load(os.path.join(script_directory, f"configs/ad_unet_config.yaml")) - from diffusers.loaders.single_file_utils import (convert_ldm_vae_checkpoint, convert_ldm_unet_checkpoint, create_text_encoder_from_ldm_clip_checkpoint, create_vae_diffusers_config, create_unet_diffusers_config) + from .single_file_utils import (convert_ldm_vae_checkpoint, convert_ldm_unet_checkpoint, create_text_encoder_from_ldm_clip_checkpoint, create_vae_diffusers_config, create_unet_diffusers_config) tokenizer = CLIPTokenizer.from_pretrained("openai/clip-vit-large-patch14")