Make RemoteMixtureOfExperts timeouts more general #125
Description
As of now, forward/backward_timeout arguments correspond only to timeouts for Server interactions. However, this is not the only possible cause of freezes: for example, beam search might take too long to finish during the forward pass. To fix this, we need to make both timeouts more general (that is, they should measure the execution time of the entire function)