diff --git a/finetrainers/cogvideox/__init__.py b/finetrainers/cogvideox/__init__.py index 9a707b8..8ca65b7 100644 --- a/finetrainers/cogvideox/__init__.py +++ b/finetrainers/cogvideox/__init__.py @@ -1,2 +1 @@ -from .cogvideox_lora import COGVIDEOX_T2V_LORA_CONFIG -from .cogvideox_lora import COGVIDEOX_T2V_FULL_FINETUNE_CONFIG \ No newline at end of file +from .cogvideox_lora import COGVIDEOX_T2V_FULL_FINETUNE_CONFIG, COGVIDEOX_T2V_LORA_CONFIG diff --git a/finetrainers/hunyuan_video/__init__.py b/finetrainers/hunyuan_video/__init__.py index 1ee2837..1800b2e 100644 --- a/finetrainers/hunyuan_video/__init__.py +++ b/finetrainers/hunyuan_video/__init__.py @@ -1,2 +1 @@ -from .hunyuan_video_lora import HUNYUAN_VIDEO_T2V_LORA_CONFIG -from .hunyuan_video_lora import HUNYUAN_VIDEO_T2V_FULL_FINETUNE_CONFIG \ No newline at end of file +from .hunyuan_video_lora import HUNYUAN_VIDEO_T2V_FULL_FINETUNE_CONFIG, HUNYUAN_VIDEO_T2V_LORA_CONFIG diff --git a/finetrainers/ltx_video/__init__.py b/finetrainers/ltx_video/__init__.py index 0476fc6..0b33512 100644 --- a/finetrainers/ltx_video/__init__.py +++ b/finetrainers/ltx_video/__init__.py @@ -1,2 +1 @@ -from .ltx_video_lora import LTX_VIDEO_T2V_LORA_CONFIG -from .ltx_video_lora import LTX_VIDEO_T2V_FULL_FINETUNE_CONFIG \ No newline at end of file +from .ltx_video_lora import LTX_VIDEO_T2V_FULL_FINETUNE_CONFIG, LTX_VIDEO_T2V_LORA_CONFIG diff --git a/finetrainers/models.py b/finetrainers/models.py index d3ece75..c24ab95 100644 --- a/finetrainers/models.py +++ b/finetrainers/models.py @@ -1,8 +1,9 @@ from typing import Any, Dict -from .cogvideox import COGVIDEOX_T2V_LORA_CONFIG, COGVIDEOX_T2V_FULL_FINETUNE_CONFIG -from .hunyuan_video import HUNYUAN_VIDEO_T2V_LORA_CONFIG, HUNYUAN_VIDEO_T2V_FULL_FINETUNE_CONFIG -from .ltx_video import LTX_VIDEO_T2V_LORA_CONFIG, LTX_VIDEO_T2V_FULL_FINETUNE_CONFIG +from .cogvideox import COGVIDEOX_T2V_FULL_FINETUNE_CONFIG, COGVIDEOX_T2V_LORA_CONFIG +from .hunyuan_video import HUNYUAN_VIDEO_T2V_FULL_FINETUNE_CONFIG, HUNYUAN_VIDEO_T2V_LORA_CONFIG +from .ltx_video import LTX_VIDEO_T2V_FULL_FINETUNE_CONFIG, LTX_VIDEO_T2V_LORA_CONFIG + SUPPORTED_MODEL_CONFIGS = { "hunyuan_video": {