add compatibility of distributedwrapper for two stream input

[NoTody]

Add compatibility of distributedwrapper for two stream input, so wrapped miner and loss can also use forward as self.miner(query_embed, labels, doc_embed, labels.clone()) and self.loss(query_embed, labels, doc_embed, labels.clone())