mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-03-26 01:21:23 +08:00
9 lines
332 B
Python
9 lines
332 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from torch.nn.parallel import DataParallel, DistributedDataParallel
|
|
|
|
from annotator.uniformer.mmcv.utils import Registry
|
|
|
|
MODULE_WRAPPERS = Registry('module wrapper')
|
|
MODULE_WRAPPERS.register_module(module=DataParallel)
|
|
MODULE_WRAPPERS.register_module(module=DistributedDataParallel)
|