Spaces:

Robert001
/

UniControl-Demo

Runtime error

first commit

b334e29 12 months ago

No virus

332 Bytes

	# 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)