298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-02-28 14:40:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove useless code.

Browse Source
This commit is contained in:
comfyanonymous 2025-01-13 03:52:37 -05:00
parent 90f349f93d
commit 1f1c7b7b56
1 changed files with 1 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
comfy/ldm/cosmos/cosmos_tokenizer/utils.py
View File

@ -17,7 +17,7 @@
from typing import Any
import torch
from einops import pack, rearrange, unpack
from einops import rearrange
import comfy.ops
@ -98,14 +98,6 @@ def default(*args):
return None
def pack_one(t, pattern):
return pack([t], pattern)
def unpack_one(t, ps, pattern):
return unpack(t, ps, pattern)[0]
def round_ste(z: torch.Tensor) -> torch.Tensor:
"""Round with straight through gradients."""
zhat = z.round()