From e4fe24ee27281a74174f2a5b12635d841ab6263d Mon Sep 17 00:00:00 2001 From: Scorpinaus Date: Mon, 3 Jun 2024 22:14:06 +0800 Subject: [PATCH] Updated categories --- sd15_clip_loader.py | 2 +- sd15_combined_loader.py | 2 +- sd15_unet_loader.py | 2 +- sd15_vae_loader.py | 2 +- sdxl_clip_loader.py | 2 +- sdxl_combined_loader.py | 2 +- sdxl_unet_loader.py | 2 +- sdxl_vae_loader.py | 2 +- 8 files changed, 8 insertions(+), 8 deletions(-) diff --git a/sd15_clip_loader.py b/sd15_clip_loader.py index 39ca341..3323d3b 100644 --- a/sd15_clip_loader.py +++ b/sd15_clip_loader.py @@ -23,7 +23,7 @@ class SD15CLIPLoader: RETURN_TYPES = ("CLIP",) FUNCTION = "load_clip" - CATEGORY = "testing_nodes" + CATEGORY = "DiffusersLoader/SD1.5" def load_clip(self, sub_directory, clip_type="stable_diffusion"): # Determine the clip type diff --git a/sd15_combined_loader.py b/sd15_combined_loader.py index f82a58b..2ae6fc4 100644 --- a/sd15_combined_loader.py +++ b/sd15_combined_loader.py @@ -18,7 +18,7 @@ class CombinedDiffusersSD15Loader: RETURN_TYPES = ("CLIP", "MODEL", "VAE") FUNCTION = "load_models" - CATEGORY = "testing_nodes" + CATEGORY = "DiffusersLoader/SD1.5" def load_models(self, sub_directory, clip_type="stable_diffusion"): clip = self.load_clip(sub_directory, clip_type) diff --git a/sd15_unet_loader.py b/sd15_unet_loader.py index 63017f9..e203776 100644 --- a/sd15_unet_loader.py +++ b/sd15_unet_loader.py @@ -38,7 +38,7 @@ class SD15UNETLoader: RETURN_TYPES = ("MODEL",) FUNCTION = "load_unet" - CATEGORY = "testing_nodes" + CATEGORY = "DiffusersLoader/SD1.5" def load_unet(self, sub_directory): unet_folder = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', 'models', 'diffusers', 'SD15', sub_directory, 'unet')) diff --git a/sd15_vae_loader.py b/sd15_vae_loader.py index 0c00d2a..f1ea65d 100644 --- a/sd15_vae_loader.py +++ b/sd15_vae_loader.py @@ -35,7 +35,7 @@ class SD15VAELoader: RETURN_TYPES = ("VAE",) FUNCTION = "load_vae" - CATEGORY = "testing_nodes" + CATEGORY = "DiffusersLoader/SD1.5" def load_vae(self, sub_directory): base_path = self.get_base_path() diff --git a/sdxl_clip_loader.py b/sdxl_clip_loader.py index 17d426f..4c5ff08 100644 --- a/sdxl_clip_loader.py +++ b/sdxl_clip_loader.py @@ -21,7 +21,7 @@ class SDXLCLIPLoader: RETURN_TYPES = ("CLIP",) FUNCTION = "load_sdxl_clip" - CATEGORY = "testing_nodes" + CATEGORY = "DiffusersLoader/SDXL" def load_sdxl_clip(self, sub_directory): script_dir = os.path.dirname(os.path.realpath(__file__)) diff --git a/sdxl_combined_loader.py b/sdxl_combined_loader.py index 1f41cf7..acfb50a 100644 --- a/sdxl_combined_loader.py +++ b/sdxl_combined_loader.py @@ -34,7 +34,7 @@ class CombinedDiffusersSDXLLoader: RETURN_TYPES = ("MODEL", "CLIP", "VAE") FUNCTION = "load_models" - CATEGORY = "testing_nodes" + CATEGORY = "DiffusersLoader/SDXL" def load_models(self, sub_directory): base_path = self.get_base_path() diff --git a/sdxl_unet_loader.py b/sdxl_unet_loader.py index 1bd8aea..f96a97e 100644 --- a/sdxl_unet_loader.py +++ b/sdxl_unet_loader.py @@ -38,7 +38,7 @@ class SDXLUNETLoader: RETURN_TYPES = ("MODEL",) FUNCTION = "load_unet" - CATEGORY = "testing_nodes" + CATEGORY = "DiffusersLoader/SDXL" def load_unet(self, sub_directory): unet_folder = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', 'models', 'diffusers', 'SDXL', sub_directory, 'unet')) diff --git a/sdxl_vae_loader.py b/sdxl_vae_loader.py index 9059d53..e25e283 100644 --- a/sdxl_vae_loader.py +++ b/sdxl_vae_loader.py @@ -35,7 +35,7 @@ class SDXLVAELoader: RETURN_TYPES = ("VAE",) FUNCTION = "load_vae" - CATEGORY = "testing_nodes" + CATEGORY = "DiffusersLoader/SDXL" def load_vae(self, sub_directory): base_path = self.get_base_path()