From ebe5b319e7132f2cbdc6dd9e7f9bf544c9c36662 Mon Sep 17 00:00:00 2001 From: Yury Gitman Date: Fri, 24 May 2024 19:46:48 -0700 Subject: [PATCH] Make torchaudio loading optional when compiling for AMD Summary: TSIA Differential Revision: D57803200 fbshipit-source-id: e45fef35243258cccd41b8bdca6b0893fab94a3d --- mmf/datasets/processors/image_processors.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/mmf/datasets/processors/image_processors.py b/mmf/datasets/processors/image_processors.py index 22d3b6d82..314d39531 100644 --- a/mmf/datasets/processors/image_processors.py +++ b/mmf/datasets/processors/image_processors.py @@ -41,12 +41,13 @@ def __init__(self, config, *args, **kwargs): transform = getattr(transforms, transform_type, None) if transform is None: - from mmf.utils.env import setup_torchaudio + if not (torch.cuda.is_available() and torch.version.hip): + from mmf.utils.env import setup_torchaudio - setup_torchaudio() - from torchaudio import transforms as torchaudio_transforms + setup_torchaudio() + from torchaudio import transforms as torchaudio_transforms - transform = getattr(torchaudio_transforms, transform_type, None) + transform = getattr(torchaudio_transforms, transform_type, None) # If torchvision or torchaudiodoesn't contain this, check our registry # if we implemented a custom transform as processor if transform is None: