Fix bug where LLama 13B wasn't able to properly run

On LLama 13B, which runs on 2GPU's, there are multiple processes, one
per GPU. However, only the first rank is supposed to communicate with
the user via keyboard + screen. This wasn't the case.