xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-09 04:44:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add LoadResAdapterNormalization

Browse Source 
For loading the normalization unet for: https://github.com/bytedance/res-adapter
This commit is contained in:
Kijai 2024-03-14 17:36:24 +02:00
parent df86fd77ac
commit 9a6aaa6518
1 changed files with 25 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
nodes.py
View File

@ -3797,6 +3797,28 @@ class RemapMaskRange:
return (scaled_mask, )
class LoadResAdapterNormalization:
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"model": ("MODEL",),
"resadapter_path": (folder_paths.get_filename_list("checkpoints"), )
}
}
RETURN_TYPES = ("MODEL",)
FUNCTION = "load_res_adapter"
def load_res_adapter(self, model, resadapter_path):
resadapter_path = f"{folder_paths.get_folder_paths('checkpoints')[0]}/{resadapter_path}"
if os.path.exists(resadapter_path):
norm_state_dict = comfy.utils.load_torch_file(resadapter_path)
model_clone = model.clone()
model_clone.model.load_state_dict(norm_state_dict, strict=False)
return (model_clone, )
NODE_CLASS_MAPPINGS = {
"INTConstant": INTConstant,
"FloatConstant": FloatConstant,
@ -3866,7 +3888,8 @@ NODE_CLASS_MAPPINGS = {
"EffnetEncode": EffnetEncode,
"ImageNormalize_Neg1_To_1": ImageNormalize_Neg1_To_1,
"Intrinsic_lora_sampling": Intrinsic_lora_sampling,
"RemapMaskRange": RemapMaskRange
"RemapMaskRange": RemapMaskRange,
"LoadResAdapterNormalization": LoadResAdapterNormalization
}
NODE_DISPLAY_NAME_MAPPINGS = {
"INTConstant": "INT Constant",
@ -3937,4 +3960,5 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"ImageNormalize_Neg1_To_1": "ImageNormalize_Neg1_To_1",
"Intrinsic_lora_sampling": "Intrinsic_lora_sampling",
"RemapMaskRange": "RemapMaskRange",
"LoadResAdapterNormalization": "LoadResAdapterNormalization",
}