Function print_rank_0

deepspeed-telechat/utils/utils.py:14–16

(msg, rank=0)

import shutil

def print_rank_0(msg, rank=0):
    if rank <= 0:
        print(msg)


def to_device(batch, device):