imform-dev
/

stable-diffusion-webui

Model card Files Files and versions Community

stable-diffusion-webui / extensions /sd-webui-controlnet /annotator /mmpkg /mmcv /parallel /registry.py

방재호

init

5636c1c over 1 year ago

history blame contribute delete

328 Bytes

	# Copyright (c) OpenMMLab. All rights reserved.
	from torch.nn.parallel import DataParallel, DistributedDataParallel

	from annotator.mmpkg.mmcv.utils import Registry

	MODULE_WRAPPERS = Registry('module wrapper')
	MODULE_WRAPPERS.register_module(module=DataParallel)
	MODULE_WRAPPERS.register_module(module=DistributedDataParallel)