andreas/ComfyUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix canny node not working with fp16. (#13085)

Browse Source
This commit is contained in:
comfyanonymous
2026-03-20 20:15:50 -07:00
committed by GitHub
parent dc719cde9c
commit a11f68dd3b
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comfy_extras/nodes_canny.py
+3 -2
View File
@@ -3,6 +3,7 @@ from typing_extensions import override
import comfy.model_management import comfy.model_management
from comfy_api.latest import ComfyExtension, io from comfy_api.latest import ComfyExtension, io
import torch
class Canny(io.ComfyNode): class Canny(io.ComfyNode):
@@ -29,8 +30,8 @@ class Canny(io.ComfyNode):
@classmethod @classmethod
def execute(cls, image, low_threshold, high_threshold) -> io.NodeOutput: def execute(cls, image, low_threshold, high_threshold) -> io.NodeOutput:
output = canny(image.to(comfy.model_management.get_torch_device()).movedim(-1, 1), low_threshold, high_threshold) output = canny(image.to(device=comfy.model_management.get_torch_device(), dtype=torch.float32).movedim(-1, 1), low_threshold, high_threshold)
img_out = output[1].to(comfy.model_management.intermediate_device()).repeat(1, 3, 1, 1).movedim(1, -1) img_out = output[1].to(device=comfy.model_management.intermediate_device(), dtype=comfy.model_management.intermediate_dtype()).repeat(1, 3, 1, 1).movedim(1, -1)
return io.NodeOutput(img_out) return io.NodeOutput(img_out)