File tree Expand file tree Collapse file tree 2 files changed +12
-2
lines changed Expand file tree Collapse file tree 2 files changed +12
-2
lines changed Original file line number Diff line number Diff line change 11# Original from: https://github.com/ace-step/ACE-Step/blob/main/music_dcae/music_dcae_pipeline.py
22import torch
33from .autoencoder_dc import AutoencoderDC
4- import torchaudio
4+ import logging
5+ try :
6+ import torchaudio
7+ except :
8+ logging .warning ("torchaudio missing, ACE model will be broken" )
9+
510import torchvision .transforms as transforms
611from .music_vocoder import ADaMoSHiFiGANV1
712
Original file line number Diff line number Diff line change 22import torch
33import torch .nn as nn
44from torch import Tensor
5- from torchaudio .transforms import MelScale
5+ import logging
6+ try :
7+ from torchaudio .transforms import MelScale
8+ except :
9+ logging .warning ("torchaudio missing, ACE model will be broken" )
10+
611import comfy .model_management
712
813class LinearSpectrogram (nn .Module ):
You can’t perform that action at this time.
0 commit comments