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

Move nodes_chroma_radiance

Browse Source
This commit is contained in:
doctorpangloss 2025-09-22 14:33:24 -07:00
parent fd6e5c2c8d
commit 66cf9b41f2
2 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy_api/latest/__init__.py
View File

@ -121,4 +121,5 @@ __all__ = [
"InputImpl",
"Types",
"ComfyExtension",
"io",
]

24
comfy_extras/nodes_chroma_radiance.py → comfy_extras/nodes/nodes_chroma_radiance.py
View File

@ -4,9 +4,9 @@ from typing import Callable
import torch
import comfy.model_management
from comfy.nodes.common import MAX_RESOLUTION
from comfy_api.latest import ComfyExtension, io
import nodes
class EmptyChromaRadianceLatentImage(io.ComfyNode):
@classmethod
@ -15,17 +15,17 @@ class EmptyChromaRadianceLatentImage(io.ComfyNode):
node_id="EmptyChromaRadianceLatentImage",
category="latent/chroma_radiance",
inputs=[
io.Int.Input(id="width", default=1024, min=16, max=nodes.MAX_RESOLUTION, step=16),
io.Int.Input(id="height", default=1024, min=16, max=nodes.MAX_RESOLUTION, step=16),
io.Int.Input(id="width", default=1024, min=16, max=MAX_RESOLUTION, step=16),
io.Int.Input(id="height", default=1024, min=16, max=MAX_RESOLUTION, step=16),
io.Int.Input(id="batch_size", default=1, min=1, max=4096),
],
outputs=[io.Latent().Output()],
)
@classmethod
def execute(cls, *, width: int, height: int, batch_size: int=1) -> io.NodeOutput:
def execute(cls, *, width: int, height: int, batch_size: int = 1) -> io.NodeOutput:
latent = torch.zeros((batch_size, 3, height, width), device=comfy.model_management.intermediate_device())
return io.NodeOutput({"samples":latent})
return io.NodeOutput({"samples": latent})
class ChromaRadianceOptions(io.ComfyNode):
@ -68,13 +68,13 @@ class ChromaRadianceOptions(io.ComfyNode):
@classmethod
def execute(
cls,
*,
model: io.Model.Type,
preserve_wrapper: bool,
start_sigma: float,
end_sigma: float,
nerf_tile_size: int,
cls,
*,
model: io.Model.Type,
preserve_wrapper: bool,
start_sigma: float,
end_sigma: float,
nerf_tile_size: int,
) -> io.NodeOutput:
radiance_options = {}
if nerf_tile_size >= 0: