From 9ca58e9aa2accf8fa3cd7209fd5266d6775bae5b Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Tue, 7 Apr 2026 15:11:50 -0600 Subject: [PATCH] Fixed offload and quantize order of ltx 2.3 text encoder. --- extensions_built_in/diffusion_models/ltx2/ltx2.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/extensions_built_in/diffusion_models/ltx2/ltx2.py b/extensions_built_in/diffusion_models/ltx2/ltx2.py index 3685475..cb1465c 100644 --- a/extensions_built_in/diffusion_models/ltx2/ltx2.py +++ b/extensions_built_in/diffusion_models/ltx2/ltx2.py @@ -424,6 +424,12 @@ class LTX2Model(BaseModel): # remove the vision tower text_encoder.model.vision_tower = None flush() + + if self.model_config.quantize_te: + self.print_and_status_update("Quantizing Text Encoder") + quantize(text_encoder, weights=get_qtype(self.model_config.qtype_te)) + freeze(text_encoder) + flush() if ( self.model_config.layer_offloading @@ -441,12 +447,6 @@ class LTX2Model(BaseModel): text_encoder.to(self.device_torch, dtype=dtype) flush() - if self.model_config.quantize_te: - self.print_and_status_update("Quantizing Text Encoder") - quantize(text_encoder, weights=get_qtype(self.model_config.qtype_te)) - freeze(text_encoder) - flush() - self.print_and_status_update("Loading VAEs and other components") if combined_state_dict is not None: original_vae_ckpt = get_model_state_dict_from_combined_ckpt(