mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-04-18 14:32:49 +08:00
Ace step empty latent nodes follow intermediate dtype.
This commit is contained in:
parent
7083484a48
commit
0fecf6f60c
@ -80,7 +80,7 @@ class EmptyAceStepLatentAudio(io.ComfyNode):
|
|||||||
@classmethod
|
@classmethod
|
||||||
def execute(cls, seconds, batch_size) -> io.NodeOutput:
|
def execute(cls, seconds, batch_size) -> io.NodeOutput:
|
||||||
length = int(seconds * 44100 / 512 / 8)
|
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"})
|
return io.NodeOutput({"samples": latent, "type": "audio"})
|
||||||
|
|
||||||
|
|
||||||
@ -103,7 +103,7 @@ class EmptyAceStep15LatentAudio(io.ComfyNode):
|
|||||||
@classmethod
|
@classmethod
|
||||||
def execute(cls, seconds, batch_size) -> io.NodeOutput:
|
def execute(cls, seconds, batch_size) -> io.NodeOutput:
|
||||||
length = round((seconds * 48000 / 1920))
|
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"})
|
return io.NodeOutput({"samples": latent, "type": "audio"})
|
||||||
|
|
||||||
class ReferenceAudio(io.ComfyNode):
|
class ReferenceAudio(io.ComfyNode):
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user