ниже приведен мой код, которым я являюсь пытаюсь использовать.
`
импортировать ОС
из tqdm импортировать tqdm
импортировать torch
из demucs импортировать предварительно обученный
из demucs .apply import apply_model
import torchaudio
defdependent_vocals(input_folder, output_folder):
если нет os.path.exists(output_folder):
os. makedirs(output_folder)
Код: Выделить всё
wav_files = [f for f in os.listdir(input_folder) if f.endswith('.wav')]
model = pretrained.get_model('htdemucs')
model.to('cuda' if torch.cuda.is_available() else 'cpu')
for wav_file in tqdm(wav_files, desc="Processing files"):
input_path = os.path.join(input_folder, wav_file)
wav, sr = torchaudio.load(input_path)
wav = wav.to('cuda' if torch.cuda.is_available() else 'cpu')
sources = apply_model(model, wav, shifts=1, split=True, overlap=0.25)
vocals = sources[0]
output_path = os.path.join(output_folder, wav_file)
torchaudio.save(output_path, vocals.cpu(), sr)
На экране появляется следующая ошибка.
Код: Выделить всё
Traceback (most recent call last): File "/workspace/all_process.py", line 4, in from demucs import pretrained File "/usr/local/lib/python3.10/dist-packages/demucs/pretrained.py", line 13, in from dora.log import fatal, bold File "/usr/local/lib/python3.10/dist-packages/dora/__init__.py", line 66, in from .explore import Explorer, Launcher File "/usr/local/lib/python3.10/dist-packages/dora/explore.py", line 27, in from .shep import Shepherd, Sheep File "/usr/local/lib/python3.10/dist-packages/dora/shep.py", line 23, in from . import git_save File "/usr/local/lib/python3.10/dist-packages/dora/git_save.py", line 16, in from .main import DecoratedMain File "/usr/local/lib/python3.10/dist-packages/dora/main.py", line 25, in from .names import NamesMixin File "/usr/local/lib/python3.10/dist-packages/dora/names.py", line 11, in from .xp import XP File "/usr/local/lib/python3.10/dist-packages/dora/xp.py", line 15, in from .link import Link File "/usr/local/lib/python3.10/dist-packages/dora/link.py", line 13, in from . import utils File "/usr/local/lib/python3.10/dist-packages/dora/utils.py", line 18, in from omegaconf.basecontainer import BaseContainer ModuleNotFoundError: No module named 'omegaconf.basecontainer'
Я также попробовал установить модуль omegaconf как
Код: Выделить всё
pip install omegaconf
Подробнее здесь: https://stackoverflow.com/questions/787 ... -no-module