fix any
This commit is contained in:
parent
405607f616
commit
4b891f103c
20
nodes.py
20
nodes.py
@ -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)"
|
||||
}
|
Loading…
Reference in New Issue
Block a user