No module named torch nn attention.

No module named torch nn attention nn接口有ModuleDict模块 0. signal import temporal_signal_split from torch_geometric_temporal. half or torch. _flex_attention import _flex_attention: from triton. sdpa_kernel(torch. "Cannot import C:\Users\dani\SD\COMFYUI\ComfyUI\custom_nodes\ComfyUI-MochiWrapper module for custom nodes: No module named 'torch. EDIT: Fixed with - run this command in the python_embeded folder: 入门. 9 and torch 2. py", line 28, in <module> import torch. flex_attention import create_block_mask def causal (b, h, q_idx, kv_idx): return q_idx >= kv_idx # Because the sparsity pattern is independent of batch and heads, we'll set them to None (which broadcasts them) block_mask = create_block_mask (causal, B = None, H = None, Q_LEN = 1024, KV_LEN = 1024) # In this case, we don Phi3 attention module using torch. bnt zvdqww gypks nfde mlsjj azuh kwdsal ezbeu qfci tpeh vjfk fqdbfd jgh lttf ezc