Initialize the Diffusion loss module.
(self, args, device)
source not stored for this graph (policy: none)
nothing calls this directly
no test coverage detected