From 1f1c7b7b5673fac3d3d38a1291ed1171f6cdc3eb Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Mon, 13 Jan 2025 03:52:37 -0500 Subject: [PATCH] Remove useless code. --- comfy/ldm/cosmos/cosmos_tokenizer/utils.py | 10 +--------- 1 file changed, 1 insertion(+), 9 deletions(-) diff --git a/comfy/ldm/cosmos/cosmos_tokenizer/utils.py b/comfy/ldm/cosmos/cosmos_tokenizer/utils.py index 64dd5e288a0..3af8d0d0571 100644 --- a/comfy/ldm/cosmos/cosmos_tokenizer/utils.py +++ b/comfy/ldm/cosmos/cosmos_tokenizer/utils.py @@ -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()