Audio second setting in EmptyLatentAudio.

This commit is contained in:
comfyanonymous 2024-06-28 02:55:36 -04:00
parent 0d9009c96e
commit 7ecb2ec169
1 changed files with 4 additions and 3 deletions

View File

@ -14,15 +14,16 @@ class EmptyLatentAudio:
@classmethod
def INPUT_TYPES(s):
return {"required": {}}
return {"required": {"seconds": ("FLOAT", {"default": 47.6, "min": 1.0, "max": 1000.0, "step": 0.1})}}
RETURN_TYPES = ("LATENT",)
FUNCTION = "generate"
CATEGORY = "_for_testing/audio"
def generate(self):
def generate(self, seconds):
batch_size = 1
latent = torch.zeros([batch_size, 64, 1024], device=self.device)
length = round((seconds * 44100 / 2048) / 2) * 2
latent = torch.zeros([batch_size, 64, length], device=self.device)
return ({"samples":latent, "type": "audio"}, )
class VAEEncodeAudio: