New ComfyUI update broken the repo - ImportError: cannot import name 'precompute_freqs_cis' from 'comfy.ldm.lightricks.model'

[FurkanGozukara]

Traceback (most recent call last):
  File "F:\Comfy_UI_V52\ComfyUI\nodes.py", line 2155, in load_custom_node
    module_spec.loader.exec_module(module)
  File "<frozen importlib._bootstrap_external>", line 883, in exec_module
  File "<frozen importlib._bootstrap>", line 241, in _call_with_frames_removed
  File "F:\Comfy_UI_V52\ComfyUI\custom_nodes\ComfyUI-TeaCache\__init__.py", line 1, in <module>
    from .nodes import NODE_CLASS_MAPPINGS as NODES_CLASS, NODE_DISPLAY_NAME_MAPPINGS as NODES_DISPLAY
  File "F:\Comfy_UI_V52\ComfyUI\custom_nodes\ComfyUI-TeaCache\nodes.py", line 12, in <module>
    from comfy.ldm.lightricks.model import precompute_freqs_cis
ImportError: cannot import name 'precompute_freqs_cis' from 'comfy.ldm.lightricks.model' (F:\Comfy_UI_V52\ComfyUI\comfy\ldm\lightricks\model.py)

Cannot import F:\Comfy_UI_V52\ComfyUI\custom_nodes\ComfyUI-TeaCache module for custom nodes: cannot import name 'precompute_freqs_cis' from 'comfy.ldm.lightricks.model' (F:\Comfy_UI_V52\ComfyUI\comfy\ldm\lightricks\model.py)

[lukymole]

same issue,anyone can help?

[alexgolesh]

same issue

[o-l-l-i]

Same issue here too.

[kexul]

If you're not using the Lighttricks model, you can simply comment out this line.

[Evs91]

if anyone wants to check my pull request #179 - I don't do a lot with LTX so ymmv.

[BobMind758]

ComfyUI 在版本迭代中，将 precompute_freqs_cis 函数从 comfy.ldm.lightricks.model 迁移到了 comfy.ldm.modules.attention 路径下，而这些自定义节点还停留在旧的导入逻辑上，因此批量触发了导入错误。
直接在ComfyUI\comfy\ldm\lightricks\model.py 中添加兼容代码，让旧路径能映射到新的函数位置

兼容补丁：将新路径的函数映射到旧路径，解决自定义节点导入问题

try:
from comfy.ldm.modules.attention import precompute_freqs_cis
except ImportError:
# 备用实现（防止极端情况）
import torch
def precompute_freqs_cis(dim: int, end: int, theta: float = 10000.0):
freqs = 1.0 / (theta ** (torch.arange(0, dim, 2)[: (dim // 2)].float() / dim))
t = torch.arange(end, device=freqs.device)
freqs = torch.outer(t, freqs).float()
freqs_cis = torch.polar(torch.ones_like(freqs), freqs)
return freqs_cis

以下是原文件的原有代码（不要删除）

... 原文件内容 ...

[x5nder]

Compatibility patch: Maps functions from the new path to the old path, resolving issues with importing custom nodes.

try: from comfy.ldm.modules.attention import precompute_freqs_cis except ImportError: # 备用实现（防止极端情况） import torch def precompute_freqs_cis(dim: int, end: int, theta: float = 10000.0): freqs = 1.0 / (theta ** (torch.arange(0, dim, 2)[: (dim // 2)].float() / dim)) t = torch.arange(end, device=freqs.device) freqs = torch.outer(t, freqs).float() freqs_cis = torch.polar(torch.ones_like(freqs), freqs) return freqs_cis

I'm slightly confused by your code change as you didn't apply any formatting nor gave instructions on where exactly to apply them; could you just attach your model.py file to your comment?

[lepiai]

将nodes.py中12行替换为from comfy.text_encoders.llama import precompute_freqs_cis，或许是一个临时解决方案

[HarryPsalms]

same issue

[Evs91]

Compatibility patch: Maps functions from the new path to the old path, resolving issues with importing custom nodes.

try: from comfy.ldm.modules.attention import precompute_freqs_cis except ImportError: # 备用实现（防止极端情况） import torch def precompute_freqs_cis(dim: int, end: int, theta: float = 10000.0): freqs = 1.0 / (theta ** (torch.arange(0, dim, 2)[: (dim // 2)].float() / dim)) t = torch.arange(end, device=freqs.device) freqs = torch.outer(t, freqs).float() freqs_cis = torch.polar(torch.ones_like(freqs), freqs) return freqs_cis

I'm slightly confused by your code change as you didn't apply any formatting nor gave instructions on where exactly to apply them; could you just attach your model.py file to your comment?

My apologies: I fixed my commit.

[httkl]

After replacing nodes.py with your patched version I get this:

` File "/ComfyUI/custom_nodes/teacache/init.py", line 1, in
from .nodes import NODE_CLASS_MAPPINGS as NODES_CLASS, NODE_DISPLAY_NAME_MAPPINGS as NODES_DISPLAY
File "/ComfyUI/custom_nodes/teacache/nodes.py", line 12, in
from comfy.ldm.lightricks.model.embeddings_connector import Embeddings1DConnector
ModuleNotFoundError: No module named 'comfy.ldm.lightricks.model.embeddings_connector'; 'comfy.ldm.lightricks.model' is not a package

Cannot import ~/ComfyUI/custom_nodes/teacache module for custom nodes: No module named 'comfy.ldm.lightricks.model.embeddings_connector'; 'comfy.ldm.lightricks.model' is not a package
`

[Evs91]

yeah...on the road so my bad. path should be fixed now.

…

On Wed, Jan 7, 2026 at 6:50 PM httkl ***@***.***> wrote: *httkl* left a comment (welltop-cn/ComfyUI-TeaCache#178) <#178 (comment)> After replacing nodes.py with your patched version I get this: ` File "/ComfyUI/custom_nodes/teacache/*init*.py", line 1, in from .nodes import NODE_CLASS_MAPPINGS as NODES_CLASS, NODE_DISPLAY_NAME_MAPPINGS as NODES_DISPLAY File "/ComfyUI/custom_nodes/teacache/nodes.py", line 12, in from comfy.ldm.lightricks.model.embeddings_connector import Embeddings1DConnector ModuleNotFoundError: No module named 'comfy.ldm.lightricks.model.embeddings_connector'; 'comfy.ldm.lightricks.model' is not a package Cannot import ~/ComfyUI/custom_nodes/teacache module for custom nodes: No module named 'comfy.ldm.lightricks.model.embeddings_connector'; 'comfy.ldm.lightricks.model' is not a package ` — Reply to this email directly, view it on GitHub <#178 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AB5EQLNEJT67NQNLXLICM7L4FWLVFAVCNFSM6AAAAACQWYCOG2VHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZTOMRRGMYDKOBUGE> . You are receiving this because you commented.Message ID: ***@***.***>

-- --Everett

[httkl]

yeah...on the road so my bad. path should be fixed now.
…

Thank you for the hint. With "from comfy.ldm.lightricks.embeddings_connector import Embeddings1DConnector" it seems to work now.

[alenhrp]

I'm not entirely clear on the previous replies and likes. Has it been resolved? Which solution is best?
Please advise, thank you!
没完全明白上面的几个回复和点赞，解决了嘛？哪个方案最佳呢？
请指引，谢谢！