From dead3582277dbde20f4fda581ab379bdd395462b Mon Sep 17 00:00:00 2001 From: John Pollock Date: Thu, 26 Dec 2024 14:59:16 -0600 Subject: [PATCH] Add MMAudioSampler to experimental audio model loaders in __init__.py so it stays in sync with the Model and FeatureUtils Loaders. I suspect this is because it is querying the cuda device directly. This would give the wrong answer half the time depending on how the other loaders were ran via the sequencing logic. This will sync up those cuda device queries with the patched version from MultiGPU. --- __init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/__init__.py b/__init__.py index 27b6c01..c27e385 100644 --- a/__init__.py +++ b/__init__.py @@ -43,7 +43,7 @@ TARGET_NODE_NAMES = { "LoadFluxControlNet", # x-flux-comfyui - https://github.com/XLabs-AI/x-flux-comfyui "Florence2ModelLoader", "DownloadAndLoadFlorence2Model", # ComfyUI-Florence2 - https://github.com/kijai/ComfyUI-Florence2 "LTXVLoader", # ComfyUI-LTXVideo - https://github.com/Lightricks/ComfyUI-LTXVideo - "MMAudioFeatureUtilsLoader", "MMAudioModelLoader", # ComfyUI-MMAudio - https://github.com/kijai/ComfyUI-MMAudio --EXPERIMENTAL-- + "MMAudioFeatureUtilsLoader", "MMAudioModelLoader", "MMAudioSampler", # ComfyUI-MMAudio - https://github.com/kijai/ComfyUI-MMAudio --EXPERIMENTAL-- } NODE_CLASS_MAPPINGS = {}