diff --git a/videox_fun/models/__init__.py b/videox_fun/models/__init__.py
index 2a9183c..e937a68 100755
--- a/videox_fun/models/__init__.py
+++ b/videox_fun/models/__init__.py
@@ -83,6 +83,8 @@ from .mova_model import MOVAModel
 from .mova_vae_audio import AutoencoderKLMOVAAudio
 from .qwenimage21_transformer2d import (QwenImage21KVCache,
                                         QwenImage21Transformer2DModel)
+from .qwenimage21_transformer2d_control import \
+    QwenImage21ControlTransformer2DModel
 from .qwenimage21_vae import AutoencoderKLQwenImage21
 from .qwenimage_transformer2d import QwenImageTransformer2DModel
 from .qwenimage_transformer2d_control import QwenImageControlTransformer2DModel