diff --git a/comfy/model_patcher.py b/comfy/model_patcher.py index 215784874..9b76b87b6 100644 --- a/comfy/model_patcher.py +++ b/comfy/model_patcher.py @@ -761,6 +761,8 @@ class ModelPatcher: key = "{}.{}".format(n, param) self.unpin_weight(key) self.patch_weight_to_device(key, device_to=device_to) + if comfy.model_management.is_device_cuda(device_to): + torch.cuda.synchronize() logging.debug("lowvram: loaded module regularly {} {}".format(n, m)) m.comfy_patched_weights = True