wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 06:40:48 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

make more clean code for refresh feature

Browse Source
This commit is contained in:
Dr.Lt.Data 2023-03-26 21:41:29 +09:00
parent e61be509c4
commit 220914bfb3
3 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy_extras/nodes_upscale_model.py
View File

@ -16,7 +16,7 @@ class UpscaleModelLoader:
CATEGORY = "loaders"
REFRESH_LIST = [("model_name", "upscale_models")]
REFRESH_LIST = ["model_name"]
def load_model(self, model_name):
model_path = folder_paths.get_full_path("upscale_models", model_name)

20
nodes.py
View File

@ -199,7 +199,7 @@ class CheckpointLoader:
CATEGORY = "loaders"
REFRESH_LIST = [("config_name", "configs"), ("ckpt_name", "checkpoints")]
REFRESH_LIST = ["config_name", "ckpt_name"]
def load_checkpoint(self, config_name, ckpt_name, output_vae=True, output_clip=True):
config_path = folder_paths.get_full_path("configs", config_name)
@ -216,7 +216,7 @@ class CheckpointLoaderSimple:
CATEGORY = "loaders"
REFRESH_LIST = [("ckpt_name", "checkpoints")]
REFRESH_LIST = ["ckpt_name"]
def load_checkpoint(self, ckpt_name, output_vae=True, output_clip=True):
ckpt_path = folder_paths.get_full_path("checkpoints", ckpt_name)
@ -253,7 +253,7 @@ class LoraLoader:
CATEGORY = "loaders"
REFRESH_LIST = [("lora_name", "loras")]
REFRESH_LIST = ["lora_name"]
def load_lora(self, model, clip, lora_name, strength_model, strength_clip):
lora_path = folder_paths.get_full_path("loras", lora_name)
@ -269,7 +269,7 @@ class VAELoader:
CATEGORY = "loaders"
REFRESH_LIST = [("vae_name", "vae")]
REFRESH_LIST = ["vae_name"]
#TODO: scale factor?
def load_vae(self, vae_name):
@ -287,7 +287,7 @@ class ControlNetLoader:
CATEGORY = "loaders"
REFRESH_LIST = [("control_net_name", "controlnet")]
REFRESH_LIST = ["control_net_name"]
def load_controlnet(self, control_net_name):
controlnet_path = folder_paths.get_full_path("controlnet", control_net_name)
@ -305,7 +305,7 @@ class DiffControlNetLoader:
CATEGORY = "loaders"
REFRESH_LIST = [("control_net_name", "controlnet")]
REFRESH_LIST = ["control_net_name"]
def load_controlnet(self, model, control_net_name):
controlnet_path = folder_paths.get_full_path("controlnet", control_net_name)
@ -349,7 +349,7 @@ class CLIPLoader:
CATEGORY = "loaders"
REFRESH_LIST = [("clip_name", "clip")]
REFRESH_LIST = ["clip_name"]
def load_clip(self, clip_name):
clip_path = folder_paths.get_full_path("clip", clip_name)
@ -366,7 +366,7 @@ class CLIPVisionLoader:
CATEGORY = "loaders"
REFRESH_LIST = [("clip_name", "clip_vision")]
REFRESH_LIST = ["clip_name"]
def load_clip(self, clip_name):
clip_path = folder_paths.get_full_path("clip_vision", clip_name)
@ -398,7 +398,7 @@ class StyleModelLoader:
CATEGORY = "loaders"
REFRESH_LIST = [("style_model_name", "style_models")]
REFRESH_LIST = ["style_model_name"]
def load_style_model(self, style_model_name):
style_model_path = folder_paths.get_full_path("style_models", style_model_name)
@ -834,7 +834,7 @@ class LoadImage:
RETURN_TYPES = ("IMAGE", "MASK")
FUNCTION = "load_image"
REFRESH_LIST = [("image", "input")]
REFRESH_LIST = ["image"]
def load_image(self, image):
image_path = os.path.join(self.input_dir, image)

4
web/scripts/app.js
View File

@ -816,8 +816,8 @@ class ComfyApp {
const def = defs[node.type];
for(const i in def.refresh_list) {
const item = def.refresh_list[i];
const filelist = def.input["required"][item[0]];
const w = node.widgets.find((w) => w.name === item[0]);
const filelist = def.input["required"][item];
const w = node.widgets.find((w) => w.name === item);
w.options.values = filelist[0];
w.value = w.options.values[0];
}