Allow using float fps for LTXVEmptyLatentAudio (#15106)

This commit is contained in:
Jukka Seppänen
2026-07-27 23:04:14 +03:00
committed by GitHub
parent c06ee57933
commit a3572c4832
2 changed files with 13 additions and 10 deletions

View File

@@ -185,7 +185,7 @@ class AudioVAE(torch.nn.Module):
self.autoencoder.mel_bins,
)
def num_of_latents_from_frames(self, frames_number: int, frame_rate: int) -> int:
def num_of_latents_from_frames(self, frames_number: int, frame_rate: float) -> int:
return math.ceil((float(frames_number) / frame_rate) * self.latents_per_second)
def run_vocoder(self, mel_spec: torch.Tensor) -> torch.Tensor: