Ace step empty latent nodes follow intermediate dtype. (#13313)

2026-04-30 04:12:37 +08:00 · 2026-04-06 18:12:16 -07:00 · 2026-04-06 18:12:16 -07:00 · 61c2387436
commit 61c2387436
parent 7083484a48
1 changed files with 2 additions and 2 deletions
--- a/comfy_extras/nodes_ace.py
+++ b/comfy_extras/nodes_ace.py
@ -80,7 +80,7 @@ class EmptyAceStepLatentAudio(io.ComfyNode):
    @classmethod
    def execute(cls, seconds, batch_size) -> io.NodeOutput:
        length = int(seconds * 44100 / 512 / 8)
-        latent = torch.zeros([batch_size, 8, 16, length], device=comfy.model_management.intermediate_device())
+        latent = torch.zeros([batch_size, 8, 16, length], device=comfy.model_management.intermediate_device(), dtype=comfy.model_management.intermediate_dtype())
        return io.NodeOutput({"samples": latent, "type": "audio"})


@ -103,7 +103,7 @@ class EmptyAceStep15LatentAudio(io.ComfyNode):
    @classmethod
    def execute(cls, seconds, batch_size) -> io.NodeOutput:
        length = round((seconds * 48000 / 1920))
-        latent = torch.zeros([batch_size, 64, length], device=comfy.model_management.intermediate_device())
+        latent = torch.zeros([batch_size, 64, length], device=comfy.model_management.intermediate_device(), dtype=comfy.model_management.intermediate_dtype())
        return io.NodeOutput({"samples": latent, "type": "audio"})

 class ReferenceAudio(io.ComfyNode):