facebookresearch--audiocraft
Alexandre Défossez
33 行
872 B
Python
33 行
872 B
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
|
|
from pathlib import Path
|
|
import typing as tp
|
|
|
|
import torch
|
|
import torchaudio
|
|
|
|
|
|
def get_white_noise(chs: int = 1, num_frames: int = 1):
|
|
wav = torch.randn(chs, num_frames)
|
|
return wav
|
|
|
|
|
|
def get_batch_white_noise(bs: int = 1, chs: int = 1, num_frames: int = 1):
|
|
wav = torch.randn(bs, chs, num_frames)
|
|
return wav
|
|
|
|
|
|
def save_wav(path: str, wav: torch.Tensor, sample_rate: int):
|
|
fp = Path(path)
|
|
kwargs: tp.Dict[str, tp.Any] = {}
|
|
if fp.suffix == '.wav':
|
|
kwargs['encoding'] = 'PCM_S'
|
|
kwargs['bits_per_sample'] = 16
|
|
elif fp.suffix == '.mp3':
|
|
kwargs['compression'] = 320
|
|
torchaudio.save(str(fp), wav, sample_rate, **kwargs)
|