Create __init__.py

2025-05-13 14:26:19 +08:00
parent 90d0ebf119
commit e784f6ea7d
1 changed files with 6 additions and 0 deletions
@@ -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.')