Updated categories
This commit is contained in:
+1
-1
@@ -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
|
||||
|
||||
@@ -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
@@ -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
@@ -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
@@ -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__))
|
||||
|
||||
@@ -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
@@ -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
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user