feat(utils): Optimize tiled_scale_multidim to support multiple GPU

- add DataParallel model support for tiled_scale_multidim
2026-08-14 12:43:29 +08:00 · 2025-07-03 16:54:51 +08:00 · 2025-07-03 16:54:51 +08:00 · 2f386dcba1
commit 2f386dcba1
parent a0f66a7262
1 changed files with 1 additions and 0 deletions
--- 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