xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-14 12:16:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(utils): Optimize tiled_scale_multidim to support multiple GPU

Browse Source 
- add DataParallel model support for tiled_scale_multidim
This commit is contained in:
MagicBear 2025-07-03 16:54:51 +08:00
parent a0f66a7262
commit 2f386dcba1
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/utils.py
View File

@ -27,6 +27,7 @@ from PIL import Image
import logging import logging
import itertools import itertools
from torch.nn.functional import interpolate from torch.nn.functional import interpolate
from torch.nn import DataParallel
from einops import rearrange from einops import rearrange
from comfy.cli_args import args from comfy.cli_args import args