wuyingwen/comfyui
1
0
Fork 0
Code Issues Pull Requests Actions 5 Packages Projects Releases Wiki Activity

Make torchaudio not a hard requirement. (#7987)

Browse Source 
Some platforms can't install it apparently so if it's not there it should
only break models that actually use it.
This commit is contained in:
comfyanonymous 2025-05-07 18:37:12 -07:00 committed by GitHub
parent 56b6ee6754
commit fd08e39588
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy/ldm/ace/vae/music_dcae_pipeline.py
View File

@ -1,7 +1,12 @@
# Original from: https://github.com/ace-step/ACE-Step/blob/main/music_dcae/music_dcae_pipeline.py # Original from: https://github.com/ace-step/ACE-Step/blob/main/music_dcae/music_dcae_pipeline.py
import torch import torch
from .autoencoder_dc import AutoencoderDC from .autoencoder_dc import AutoencoderDC
import torchaudio import logging
try:
import torchaudio
except:
logging.warning("torchaudio missing, ACE model will be broken")
import torchvision.transforms as transforms import torchvision.transforms as transforms
from .music_vocoder import ADaMoSHiFiGANV1 from .music_vocoder import ADaMoSHiFiGANV1

7
comfy/ldm/ace/vae/music_log_mel.py
View File

@ -2,7 +2,12 @@
import torch import torch
import torch.nn as nn import torch.nn as nn
from torch import Tensor from torch import Tensor
from torchaudio.transforms import MelScale import logging
try:
from torchaudio.transforms import MelScale
except:
logging.warning("torchaudio missing, ACE model will be broken")
import comfy.model_management import comfy.model_management
class LinearSpectrogram(nn.Module): class LinearSpectrogram(nn.Module):