From 2f386dcba1857ec89500b95ccf13eea75792b8da Mon Sep 17 00:00:00 2001
From: MagicBear <mb@bilibili.com>
Date: Thu, 3 Jul 2025 16:54:51 +0800
Subject: [PATCH] feat(utils): Optimize tiled_scale_multidim to support
 multiple GPU

- add DataParallel model support for tiled_scale_multidim
---
 comfy/utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/comfy/utils.py b/comfy/utils.py
index c1d04145e..5e16ae761 100644
--- a/comfy/utils.py
+++ b/comfy/utils.py
@@ -27,6 +27,7 @@ from PIL import Image
 import logging
 import itertools
 from torch.nn.functional import interpolate
+from torch.nn import DataParallel
 from einops import rearrange
 from comfy.cli_args import args