Function sync_device

tests/stable_audio/stable_audio_python_warm_bench.py:225–227

(device: str)

def sync_device(device: str) -> None:
    if str(device).startswith("cuda"):
        torch.cuda.synchronize(torch.device(device))


def seed_all(seed: int) -> None: