remove redundant hijacks

This commit is contained in:
Seunghoon Lee
2024-06-16 04:59:20 +09:00
parent a673bebdef
commit 68e7692ed5
9 changed files with 1 additions and 532 deletions
-2
View File
@@ -37,8 +37,6 @@ def diffusers_callback(pipe, step: int, timestep: int, kwargs: dict):
if p is None:
return kwargs
latents = kwargs.get('latents', None)
if torch.is_tensor(latents) and latents.device.type == "privateuseone":
torch.dml.synchronize_tensor(latents) # DML synchronize
debug_callback(f'Callback: step={step} timestep={timestep} latents={latents.shape if latents is not None else None} kwargs={list(kwargs)}')
shared.state.sampling_step = step
if shared.state.interrupted or shared.state.skipped: