Function empty_device_cache

src/diffusers/utils/torch_utils.py:303–309

(device_type: str | None = None)

def empty_device_cache(device_type: str | None = None):
    if device_type is None:
        device_type = get_device()
    if device_type in ["cpu"]:
        return
    device_mod = getattr(torch, device_type, torch.cuda)
    device_mod.empty_cache()


def device_synchronize(device_type: str | None = None):