internals/util/__init__.py

import os

import torch

from internals.util.config import get_root_dir


def getcwd():
    return get_root_dir()


def get_generators(seed, num_generators=1):
    return [torch.Generator().manual_seed(seed + i) for i in range(num_generators)]