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

Update model_optimization_nodes.py

Browse Source
This commit is contained in:
kijai 2025-08-06 09:45:54 +03:00
parent 984c7bb276
commit 331260d908
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nodes/model_optimization_nodes.py
View File

@ -19,7 +19,10 @@ if not _initialized:
_original_functions["orig_attention"] = comfy_attention.optimized_attention _original_functions["orig_attention"] = comfy_attention.optimized_attention
_original_functions["original_patch_model"] = comfy.model_patcher.ModelPatcher.patch_model _original_functions["original_patch_model"] = comfy.model_patcher.ModelPatcher.patch_model
_original_functions["original_load_lora_for_models"] = comfy.sd.load_lora_for_models _original_functions["original_load_lora_for_models"] = comfy.sd.load_lora_for_models
_original_functions["original_qwen_forward"] = comfy.ldm.qwen_image.model.Attention.forward try:
_original_functions["original_qwen_forward"] = comfy.ldm.qwen_image.model.Attention.forward
except:
pass
_initialized = True _initialized = True
class BaseLoaderKJ: class BaseLoaderKJ: