This commit is contained in:
comfyanonymous
2025-09-12 18:57:04 -07:00
committed by GitHub
parent 2559dee492
commit 29bf807b0e
2 changed files with 2 additions and 2 deletions

View File

@@ -31,7 +31,7 @@ class AudioEncoderModel():
def encode_audio(self, audio, sample_rate):
comfy.model_management.load_model_gpu(self.patcher)
audio = torchaudio.functional.resample(audio, sample_rate, self.model_sample_rate)
out, all_layers = self.model(audio.to(self.load_device), sr=self.model_sample_rate)
out, all_layers = self.model(audio.to(self.load_device))
outputs = {}
outputs["encoded_audio"] = out
outputs["encoded_audio_all_layers"] = all_layers