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

View File

@ -32,24 +32,20 @@ class AudioLoadPath:
else:
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 = {
"waveform": audio_tensor.unsqueeze(0), # 添加批次维度 [batch, samples]
"sample_rate": sample_rate
}
# 构建音频字典
# audio_dict = {
# "waveform": audio,
# "sample_rate": sample_rate
# }
return (audio_dict,)
return (audio,)
except Exception as e:
raise Exception(f"加载音频失败: {str(e)}")
NODE_CLASS_MAPPINGS = {
"AudioLoadPath": AudioLoadPath,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"AudioLoadPath": "Load Audio (Path/URL)"
}