wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-22 20:30:25 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

update extra nodes

Browse Source
This commit is contained in:
doctorpangloss 2025-12-09 09:05:18 -08:00
parent 26e9b67732
commit a7ef3e04ea
4 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_patcher.py
View File

@ -340,7 +340,7 @@ class ModelPatcher(ModelManageable, PatchSupport):
def loaded_size(self): def loaded_size(self):
return self._memory_measurements.model_loaded_weight_memory return self._memory_measurements.model_loaded_weight_memory
def clone(self): def clone(self) -> "ModelPatcher":
n = self.__class__(self.model, self.load_device, self.offload_device, self.model_size(), weight_inplace_update=self.weight_inplace_update) n = self.__class__(self.model, self.load_device, self.offload_device, self.model_size(), weight_inplace_update=self.weight_inplace_update)
n._memory_measurements = self._memory_measurements n._memory_measurements = self._memory_measurements
n.ckpt_name = self.ckpt_name n.ckpt_name = self.ckpt_name

4
comfy_extras/nodes/nodes_dataset.py
View File

@ -7,8 +7,8 @@ import torch
from PIL import Image from PIL import Image
from typing_extensions import override from typing_extensions import override
import folder_paths from comfy.cmd import folder_paths
import node_helpers from comfy import node_helpers
from comfy_api.latest import ComfyExtension, io from comfy_api.latest import ComfyExtension, io

4
comfy_extras/nodes/nodes_kandinsky5.py
View File

@ -1,5 +1,5 @@
import nodes from comfy.nodes import base_nodes as nodes
import node_helpers from comfy import node_helpers
import torch import torch
import comfy.model_management import comfy.model_management
import comfy.utils import comfy.utils

3
comfy_extras/nodes/nodes_rope.py
View File

@ -1,3 +1,4 @@
from comfy.model_patcher import ModelPatcher
from comfy_api.latest import ComfyExtension, io from comfy_api.latest import ComfyExtension, io
from typing_extensions import override from typing_extensions import override
@ -29,7 +30,7 @@ class ScaleROPE(io.ComfyNode):
) )
@classmethod @classmethod
def execute(cls, model, scale_x, shift_x, scale_y, shift_y, scale_t, shift_t) -> io.NodeOutput: def execute(cls, model: ModelPatcher, scale_x, shift_x, scale_y, shift_y, scale_t, shift_t) -> io.NodeOutput:
m = model.clone() m = model.clone()
m.set_model_rope_options(scale_x, shift_x, scale_y, shift_y, scale_t, shift_t) m.set_model_rope_options(scale_x, shift_x, scale_y, shift_y, scale_t, shift_t)
return io.NodeOutput(m) return io.NodeOutput(m)