Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Force synchronize with ipex

Browse Source
This commit is contained in:
Disty0
2025-09-02 14:26:35 +03:00
parent aa9a5a678e
commit a4e51759b9
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_callbacks.py
+3 -1
View File
@@ -54,7 +54,9 @@ 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" or (devices.backend == "rocm" and sys.platform == "win32"):
if devices.backend == "ipex": # xe driver on linux needs this
torch.xpu.synchronize(devices.device)
elif devices.backend == "zluda" or (devices.backend == "rocm" and sys.platform == "win32"):
torch.cuda.synchronize(devices.device)
latents = kwargs.get('latents', None)
if debug: