298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 02:15:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Small optimization.

Browse Source
This commit is contained in:
comfyanonymous 2024-06-15 02:44:38 -04:00
parent f2e844e054
commit 1281f933c1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ldm/modules/diffusionmodules/mmdit.py
View File

@ -243,9 +243,9 @@ class TimestepEmbedder(nn.Module):
half = dim // 2
freqs = torch.exp(
-math.log(max_period)
* torch.arange(start=0, end=half, dtype=torch.float32)
* torch.arange(start=0, end=half, dtype=torch.float32, device=t.device)
/ half
).to(device=t.device)
)
args = t[:, None].float() * freqs[None]
embedding = torch.cat([torch.cos(args), torch.sin(args)], dim=-1)
if dim % 2: