From cf97c9fbe864124babd4edc7f4256c2e38d5fb1d Mon Sep 17 00:00:00 2001 From: Disty0 Date: Sat, 22 Feb 2025 23:01:44 +0300 Subject: [PATCH] Add synchronize call for zluda --- modules/processing_callbacks.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/modules/processing_callbacks.py b/modules/processing_callbacks.py index 0191eff72..b78b1e6a1 100644 --- a/modules/processing_callbacks.py +++ b/modules/processing_callbacks.py @@ -3,7 +3,7 @@ import os import time import torch import numpy as np -from modules import shared, processing_correction, extra_networks, timer, prompt_parser_diffusers +from modules import shared, devices, processing_correction, extra_networks, timer, prompt_parser_diffusers p = None @@ -51,6 +51,8 @@ def diffusers_callback_legacy(step: int, timestep: int, latents: typing.Union[to def diffusers_callback(pipe, step: int = 0, timestep: int = 0, kwargs: dict = {}): t0 = time.time() + if devices.backend == "zluda": + torch.cuda.synchronize(devices.device) if p is None: return kwargs latents = kwargs.get('latents', None)