Fix FFmpeg initialization logic

Differential Revision: D47398447 fbshipit-source-id: 34b7d63668a21234f6a9523d4f1078dd0a89aa52
pytorch · Jul 12, 2023 · e4d5364 · e4d5364
1 parent 786066b
commit e4d5364
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/torchaudio/_extension/__init__.py b/torchaudio/_extension/__init__.py
@@ -4,7 +4,11 @@
 
 from torchaudio._internal.module_utils import fail_with_message, is_module_available, no_op
 
-from .utils import _check_cuda_version, _fail_since_no_ffmpeg, _init_dll_path, _init_ffmpeg, _init_sox, _load_lib
+try:
+    from .fb import _init_ffmpeg
+except ImportError:
+    from .utils import _init_ffmpeg
+from .utils import _check_cuda_version, _fail_since_no_ffmpeg, _init_dll_path, _init_sox, _load_lib
 
 _LG = logging.getLogger(__name__)