Skip to content

Commit 6e5c8a7

Browse files
committed
fix sample_euler_ancestral for MPS
1 parent 72ce8f1 commit 6e5c8a7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

k_diffusion/sampling.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ def sample_euler_ancestral(model, x, sigmas, extra_args=None, callback=None, dis
9393
sigma_down, sigma_up = get_ancestral_step(sigmas[i], sigmas[i + 1])
9494
if callback is not None:
9595
callback({'x': x, 'i': i, 'sigma': sigmas[i], 'sigma_hat': sigmas[i], 'denoised': denoised})
96-
d = to_d(x, sigmas[i], denoised)
96+
d = to_d(x, sigmas[i], denoised, clone_please=True)
9797
# Euler method
9898
dt = sigma_down - sigmas[i]
9999
x = x + d * dt

0 commit comments

Comments
 (0)