From beae2943cf72973595ca203f2b764241eb577d6d Mon Sep 17 00:00:00 2001
From: Will Lin <wlsaidhi@gmail.com>
Date: Wed, 21 Jan 2026 13:28:43 -0800
Subject: [PATCH] cleanup

---
 fastvideo/configs/pipelines/registry.py    | 7 -------
 fastvideo/models/audio/ltx2_audio_vae.py   | 1 -
 fastvideo/models/dits/ltx2.py              | 2 +-
 fastvideo/models/vaes/ltx2vae.py           | 2 +-
 fastvideo/pipelines/basic/ltx2/__init__.py | 4 ----
 5 files changed, 2 insertions(+), 14 deletions(-)

diff --git a/fastvideo/configs/pipelines/registry.py b/fastvideo/configs/pipelines/registry.py
index e4f0f0c4e..baade6da1 100644
--- a/fastvideo/configs/pipelines/registry.py
+++ b/fastvideo/configs/pipelines/registry.py
@@ -166,13 +166,6 @@ def get_pipeline_config_cls_from_name(
 
     pipeline_config_cls: type[PipelineConfig] | None = None
 
-    # Temporary local mapping for LTX-2 diffusers-style exports.
-    if os.path.exists(pipeline_name_or_path):
-        norm_path = os.path.normpath(pipeline_name_or_path)
-        path_parts = norm_path.split(os.sep)
-        if "ltx2_diffusers" in path_parts:
-            return LTX2T2VConfig
-
     # First try exact match for specific weights
     if pipeline_name_or_path in PIPE_NAME_TO_CONFIG:
         pipeline_config_cls = PIPE_NAME_TO_CONFIG[pipeline_name_or_path]
diff --git a/fastvideo/models/audio/ltx2_audio_vae.py b/fastvideo/models/audio/ltx2_audio_vae.py
index 1f75a9848..fcc047f6c 100644
--- a/fastvideo/models/audio/ltx2_audio_vae.py
+++ b/fastvideo/models/audio/ltx2_audio_vae.py
@@ -1,7 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 """
 Native LTX-2 Audio VAE and Vocoder implementation for FastVideo.
-This is a standalone implementation that does not depend on ltx_core.
 """
 
 from __future__ import annotations
diff --git a/fastvideo/models/dits/ltx2.py b/fastvideo/models/dits/ltx2.py
index e96a97141..b8e0aecc7 100644
--- a/fastvideo/models/dits/ltx2.py
+++ b/fastvideo/models/dits/ltx2.py
@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 """
-Native LTX-2 transformer implementation (video-only).
+LTX-2 transformer implementation
 """
 
 from dataclasses import dataclass, replace
diff --git a/fastvideo/models/vaes/ltx2vae.py b/fastvideo/models/vaes/ltx2vae.py
index 6396b9519..e0a01db83 100644
--- a/fastvideo/models/vaes/ltx2vae.py
+++ b/fastvideo/models/vaes/ltx2vae.py
@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 """
-LTX-2 Video VAE implementation for FastVideo.
+LTX-2 Video VAE implementation
 """
 
 from __future__ import annotations
diff --git a/fastvideo/pipelines/basic/ltx2/__init__.py b/fastvideo/pipelines/basic/ltx2/__init__.py
index de78d8bb8..e69de29bb 100644
--- a/fastvideo/pipelines/basic/ltx2/__init__.py
+++ b/fastvideo/pipelines/basic/ltx2/__init__.py
@@ -1,4 +0,0 @@
-# SPDX-License-Identifier: Apache-2.0
-"""
-LTX-2 pipelines for FastVideo.
-"""