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)]