↓ 6 callersMethod_infer_cond_uncond(self, latents_input, prompt_embeds, infer_condition=True, txt_ids=None, img_ids=None)
lightx2v/models/networks/flux2/model.py:43
↓ 6 callersFunctioncutlass_scaled_mxfp6_mxfp8_mm(mat_a, mat_b, scales_a, scales_b, alpha, bias=None)
lightx2v_kernel/python/lightx2v_kernel/gemm.py:150
↓ 6 callersMethodinfer_block(self, weights, grid_sizes, embed, x, embed0, seq_lens, freqs, context, block_idx)
lightx2v/models/networks/wan/infer/feature_caching/transformer_infer.py:926