Updated categories

This commit is contained in:
Scorpinaus
2024-06-03 22:14:06 +08:00
parent da41ec04ef
commit e4fe24ee27
8 changed files with 8 additions and 8 deletions
+1 -1
View File
@@ -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
+1 -1
View File
@@ -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)
+1 -1
View File
@@ -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'))
+1 -1
View File
@@ -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()
+1 -1
View File
@@ -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__))
+1 -1
View File
@@ -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()
+1 -1
View File
@@ -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'))
+1 -1
View File
@@ -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()