Function is_main_process

scripts/evaluation.py:53–55

Checks if the current process is the main one (rank 0).

(rank)

def is_main_process(rank):
    """Checks if the current process is the main one (rank 0)."""
    return rank == 0


class TextPromptDataset(Dataset):