This commit is contained in:
wangbo 2025-05-15 21:11:07 +08:00
parent 405607f616
commit 4b891f103c

View File

@ -32,16 +32,16 @@ class AudioLoadPath:
else: else:
audio, _ = core.load(path, sr=sample_rate, offset=offset, duration=duration) audio, _ = core.load(path, sr=sample_rate, offset=offset, duration=duration)
# 将音频数据转换为正确的格 # 使用与参考代码相同的维度转换方
audio_tensor = torch.from_numpy(audio).float() audio = torch.from_numpy(audio)[None,:,None]
# 创建符合 SaveAudio 节点期望的字典格式 # 构建音频字典
audio_dict = { # audio_dict = {
"waveform": audio_tensor.unsqueeze(0), # 添加批次维度 [batch, samples] # "waveform": audio,
"sample_rate": sample_rate # "sample_rate": sample_rate
} # }
return (audio_dict,) return (audio,)
except Exception as e: except Exception as e:
raise Exception(f"加载音频失败: {str(e)}") raise Exception(f"加载音频失败: {str(e)}")
@ -49,7 +49,3 @@ class AudioLoadPath:
NODE_CLASS_MAPPINGS = { NODE_CLASS_MAPPINGS = {
"AudioLoadPath": AudioLoadPath, "AudioLoadPath": AudioLoadPath,
} }
NODE_DISPLAY_NAME_MAPPINGS = {
"AudioLoadPath": "Load Audio (Path/URL)"
}