MCPcopy Create free account
hub / github.com/Zhiyuan-R/Tiger-Diffusion / _transform_

Function _transform_

test_generation.py:506–507  ·  view source on GitHub ↗
(m)

Source from the content-addressed store, hash-verified

504 model.cuda()
505
506 def _transform_(m):
507 return nn.parallel.DataParallel(m)
508
509 model = model.cuda()
510 model.multi_gpu_wrapper(_transform_)

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected