n0p/mlx-video
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove commented-out code and clean up text encoder initialization

Browse Source
This commit is contained in:
Prince Canuma
2026-01-13 23:31:54 +01:00
parent ea063f7550
commit 74af04718d
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
mlx_video/models/ltx/text_encoder.py
View File

@@ -125,7 +125,6 @@ class LanguageModel(nn.Module):
sanitized = {} sanitized = {}
for key, value in weights.items(): for key, value in weights.items():
if key.startswith(prefix): if key.startswith(prefix):
# If the weight is float32, cast to bfloat16
if hasattr(value, "dtype") and value.dtype == mx.float32: if hasattr(value, "dtype") and value.dtype == mx.float32:
sanitized[key[len(prefix):]] = value.astype(mx.bfloat16) sanitized[key[len(prefix):]] = value.astype(mx.bfloat16)
else: else:
@@ -534,10 +533,8 @@ class LTX2TextEncoder(nn.Module):
text_encoder_path = str(text_encoder_path / "text_encoder") text_encoder_path = str(text_encoder_path / "text_encoder")
self.language_model = LanguageModel.from_pretrained(text_encoder_path) self.language_model = LanguageModel.from_pretrained(text_encoder_path)
print(f"Language model data type: {self.language_model.model.embed_tokens.weight.dtype}")
# Load transformer weights for feature extractor and connector # Load transformer weights for feature extractor and connector
transformer_files = list(model_path.glob("ltx-2-19*.safetensors")) transformer_files = list(model_path.glob("ltx-2-19*.safetensors"))
if transformer_files: if transformer_files:
transformer_weights = mx.load(str(transformer_files[0])) transformer_weights = mx.load(str(transformer_files[0]))