From 39994ccf2637e4579b138550375bded04dc6babf Mon Sep 17 00:00:00 2001 From: Nouamane Tazi Date: Tue, 13 Sep 2022 20:58:02 +0000 Subject: [PATCH] fix merging --- src/diffusers/models/embeddings.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/embeddings.py b/src/diffusers/models/embeddings.py index 3918a616f3..99bfa96f0d 100644 --- a/src/diffusers/models/embeddings.py +++ b/src/diffusers/models/embeddings.py @@ -35,7 +35,7 @@ def get_timestep_embedding( exponent = -math.log(max_period) * torch.arange(start=0, end=half_dim, dtype=torch.float32, device=timesteps.device) exponent = exponent / (half_dim - downscale_freq_shift) - emb = torch.exp(exponent).to(device=timesteps.device, non_blocking=True) + emb = torch.exp(exponent) emb = timesteps[:, None].float() * emb[None, :] # scale embeddings