diff --git a/matrixgame/encoder_variants/__init__.py b/matrixgame/encoder_variants/__init__.py
new file mode 100644
index 0000000..429dcd7
--- /dev/null
+++ b/matrixgame/encoder_variants/__init__.py
@@ -0,0 +1,6 @@
+from .matrixgame_i2v import MatrixGameEncoderWrapperI2V
+def get_text_enc(enc_name, model_path, weight_dtype, i2v_type):
+    if enc_name == 'matrixgame':
+        return MatrixGameEncoderWrapperI2V(model_path, weight_dtype, task = 'i2v', i2v_type = i2v_type)
+    else:
+        raise NotImplementedError(f'{enc_name} is not implemented.')