Remove useless code.

comfyanonymous · Jan 13, 2025 · 1f1c7b7 · 1f1c7b7
1 parent 90f349f
commit 1f1c7b7
Showing 1 changed file with 1 addition and 9 deletions.
diff --git 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()