Spaces:

hugging-apps
/

ugg-unified-generative-geometry

Running on Zero

ugg-unified-generative-geometry / ugg_utils /distributed_utils.py

Upload folder using huggingface_hub

705e907 verified about 5 hours ago

271 Bytes

import wandb
from typing import Callable
import torch
import torch.distributed as dist
from lightning.pytorch.utilities.rank_zero import rank_zero_only

is_rank_zero = wandb.run is not None
is_rank_zero = rank_zero_only.rank == 0

rank_zero_print = rank_zero_only(print)

	import wandb
	from typing import Callable
	import torch
	import torch.distributed as dist
	from lightning.pytorch.utilities.rank_zero import rank_zero_only

	is_rank_zero = wandb.run is not None
	is_rank_zero = rank_zero_only.rank == 0

	rank_zero_print = rank_zero_only(print)