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
flush()
else:
te_subfolder = "tokenizer"
te_tokenizer_subfolder = "tokenizer"
te_model_subfolder = "text_encoder"
if self.model_config.te_name_or_path is not None:
te_path = self.model_config.te_name_or_path
else:
if self.ltx_te_path is not None:
# pull from original repo for 2.3
te_path = self.ltx_te_path
te_subfolder = None
te_tokenizer_subfolder = ""
te_model_subfolder = ""
else:
te_path = base_te_path
tokenizer = GemmaTokenizerFast.from_pretrained(
te_path, subfolder=te_subfolder
te_path, subfolder=te_tokenizer_subfolder
)
text_encoder = Gemma3ForConditionalGeneration.from_pretrained(
te_path, subfolder=te_subfolder, dtype=dtype
te_path, subfolder=te_model_subfolder, dtype=dtype
)
# remove the vision tower