support ltx2 two stage pipeline & vram

This commit is contained in:
mi804
2026-01-30 16:55:40 +08:00
parent b1a2782ad7
commit 4f23caa55f
7 changed files with 227 additions and 85 deletions

View File

@@ -363,4 +363,4 @@ class LTX2TextEncoderPostModules(torch.nn.Module):
super().__init__()
self.feature_extractor_linear = GemmaFeaturesExtractorProjLinear()
self.embeddings_connector = Embeddings1DConnector()
self.audio_embeddings_connector = Embeddings1DConnector()
self.audio_embeddings_connector = Embeddings1DConnector()