github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix issue with using the default text encoder with ltx 2.3

Browse Source
This commit is contained in:
Jaret Burkett
2026-03-22 18:53:59 -06:00
parent 859635e95b
commit 4a3d317e2b
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
extensions_built_in/diffusion_models/ltx2/ltx2.py
View File

@@ -390,21 +390,23 @@ class LTX2Model(BaseModel):
del te_state_dict del te_state_dict
flush() flush()
else: else:
te_subfolder = "tokenizer" te_tokenizer_subfolder = "tokenizer"
te_model_subfolder = "text_encoder"
if self.model_config.te_name_or_path is not None: if self.model_config.te_name_or_path is not None:
te_path = self.model_config.te_name_or_path te_path = self.model_config.te_name_or_path
else: else:
if self.ltx_te_path is not None: if self.ltx_te_path is not None:
# pull from original repo for 2.3 # pull from original repo for 2.3
te_path = self.ltx_te_path te_path = self.ltx_te_path
te_subfolder = None te_tokenizer_subfolder = ""
te_model_subfolder = ""
else: else:
te_path = base_te_path te_path = base_te_path
tokenizer = GemmaTokenizerFast.from_pretrained( tokenizer = GemmaTokenizerFast.from_pretrained(
te_path, subfolder=te_subfolder te_path, subfolder=te_tokenizer_subfolder
) )
text_encoder = Gemma3ForConditionalGeneration.from_pretrained( text_encoder = Gemma3ForConditionalGeneration.from_pretrained(
te_path, subfolder=te_subfolder, dtype=dtype te_path, subfolder=te_model_subfolder, dtype=dtype
) )
# remove the vision tower # remove the vision tower

2
version.py
View File

@@ -1 +1 @@
VERSION = "0.7.25" VERSION = "0.7.26"