feat: add AODH Image Saver (Metadata), Lora Selector, Checkpoint Selector, and various node improvements
This commit is contained in:
9
nodes/checkpoint_from_string/__init__.py
Normal file
9
nodes/checkpoint_from_string/__init__.py
Normal file
@@ -0,0 +1,9 @@
|
||||
from .checkpoint_from_string import CheckpointLoaderFromString
|
||||
|
||||
NODE_CLASS_MAPPINGS = {
|
||||
"CheckpointLoaderFromString": CheckpointLoaderFromString
|
||||
}
|
||||
|
||||
NODE_DISPLAY_NAME_MAPPINGS = {
|
||||
"CheckpointLoaderFromString": "Checkpoint Loader (From String)"
|
||||
}
|
||||
19
nodes/checkpoint_from_string/checkpoint_from_string.py
Normal file
19
nodes/checkpoint_from_string/checkpoint_from_string.py
Normal file
@@ -0,0 +1,19 @@
|
||||
import folder_paths
|
||||
import comfy.sd
|
||||
|
||||
class CheckpointLoaderFromString:
|
||||
@classmethod
|
||||
def INPUT_TYPES(s):
|
||||
return {
|
||||
"required": {
|
||||
"ckpt_name": ("STRING", {"multiline": False, "forceInput": True}),
|
||||
}
|
||||
}
|
||||
RETURN_TYPES = ("MODEL", "CLIP", "VAE")
|
||||
FUNCTION = "load_checkpoint"
|
||||
CATEGORY = "AODH Pack"
|
||||
|
||||
def load_checkpoint(self, ckpt_name):
|
||||
ckpt_path = folder_paths.get_full_path_or_raise("checkpoints", ckpt_name)
|
||||
out = comfy.sd.load_checkpoint_guess_config(ckpt_path, output_vae=True, output_clip=True, embedding_directory=folder_paths.get_folder_paths("embeddings"))
|
||||
return out[:3]
|
||||
Reference in New Issue
Block a user