diff --git a/modules/util/LayerOffloadConductor.py b/modules/util/LayerOffloadConductor.py index 70511659d..89855c99f 100644 --- a/modules/util/LayerOffloadConductor.py +++ b/modules/util/LayerOffloadConductor.py @@ -248,7 +248,7 @@ def __init__( def reserve_cache(self, tensors: list[torch.Tensor]): num_bytes = sum(tensor.element_size() * tensor.numel() for tensor in tensors) \ - + len(tensors) * 4 # add enough padding for alignment + + len(tensors) * 16 # add enough padding for alignment if num_bytes == 0: return