From 7a8b1b1c4d75d70d9e0cca8a54cdb708ad5654c1 Mon Sep 17 00:00:00 2001 From: direwolf_xcq Date: Mon, 24 Aug 2026 14:30:52 +0800 Subject: [PATCH] [models] optimize: call text_encoder.model instead of text_encoder to skip unused LM head --- src/flow_factory/models/ltx2/ltx2_t2av.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/flow_factory/models/ltx2/ltx2_t2av.py b/src/flow_factory/models/ltx2/ltx2_t2av.py index 94ed6928e..d68c23a6a 100644 --- a/src/flow_factory/models/ltx2/ltx2_t2av.py +++ b/src/flow_factory/models/ltx2/ltx2_t2av.py @@ -357,7 +357,7 @@ def _encode_text( input_ids = tok_out.input_ids.to(device) attention_mask = tok_out.attention_mask.to(device) - enc_out = self.pipeline.text_encoder( + enc_out = self.pipeline.text_encoder.model( input_ids=input_ids, attention_mask=attention_mask, output_hidden_states=True,