diff --git a/finetrainers/cogvideox/__init__.py b/finetrainers/cogvideox/__init__.py index 390479b..7a72064 100644 --- a/finetrainers/cogvideox/__init__.py +++ b/finetrainers/cogvideox/__init__.py @@ -1,2 +1,2 @@ -from .cogvideox_lora import COGVIDEOX_T2V_LORA_CONFIG from .full_finetune import COGVIDEOX_T2V_FULL_FINETUNE_CONFIG +from .lora import COGVIDEOX_T2V_LORA_CONFIG diff --git a/finetrainers/cogvideox/full_finetune.py b/finetrainers/cogvideox/full_finetune.py index f755981..b7f2b4b 100644 --- a/finetrainers/cogvideox/full_finetune.py +++ b/finetrainers/cogvideox/full_finetune.py @@ -1,6 +1,6 @@ from diffusers import CogVideoXPipeline -from .cogvideox_lora import ( +from .lora import ( calculate_noisy_latents, collate_fn_t2v, forward_pass, diff --git a/finetrainers/cogvideox/cogvideox_lora.py b/finetrainers/cogvideox/lora.py similarity index 100% rename from finetrainers/cogvideox/cogvideox_lora.py rename to finetrainers/cogvideox/lora.py diff --git a/finetrainers/hunyuan_video/__init__.py b/finetrainers/hunyuan_video/__init__.py index e1fdafa..8ac729e 100644 --- a/finetrainers/hunyuan_video/__init__.py +++ b/finetrainers/hunyuan_video/__init__.py @@ -1,2 +1,2 @@ from .full_finetune import HUNYUAN_VIDEO_T2V_FULL_FINETUNE_CONFIG -from .hunyuan_video_lora import HUNYUAN_VIDEO_T2V_LORA_CONFIG +from .lora import HUNYUAN_VIDEO_T2V_LORA_CONFIG diff --git a/finetrainers/hunyuan_video/full_finetune.py b/finetrainers/hunyuan_video/full_finetune.py index 36dd5cb..65e73f5 100644 --- a/finetrainers/hunyuan_video/full_finetune.py +++ b/finetrainers/hunyuan_video/full_finetune.py @@ -1,6 +1,6 @@ from diffusers import HunyuanVideoPipeline -from .hunyuan_video_lora import ( +from .lora import ( collate_fn_t2v, forward_pass, initialize_pipeline, diff --git a/finetrainers/hunyuan_video/hunyuan_video_lora.py b/finetrainers/hunyuan_video/lora.py similarity index 100% rename from finetrainers/hunyuan_video/hunyuan_video_lora.py rename to finetrainers/hunyuan_video/lora.py diff --git a/finetrainers/ltx_video/__init__.py b/finetrainers/ltx_video/__init__.py index 6d5d0f9..69391cd 100644 --- a/finetrainers/ltx_video/__init__.py +++ b/finetrainers/ltx_video/__init__.py @@ -1,2 +1,2 @@ from .full_finetune import LTX_VIDEO_T2V_FULL_FINETUNE_CONFIG -from .ltx_video_lora import LTX_VIDEO_T2V_LORA_CONFIG +from .lora import LTX_VIDEO_T2V_LORA_CONFIG diff --git a/finetrainers/ltx_video/full_finetune.py b/finetrainers/ltx_video/full_finetune.py index 9aa30ef..ca799ea 100644 --- a/finetrainers/ltx_video/full_finetune.py +++ b/finetrainers/ltx_video/full_finetune.py @@ -1,6 +1,6 @@ from diffusers import LTXPipeline -from .ltx_video_lora import ( +from .lora import ( collate_fn_t2v, forward_pass, initialize_pipeline, diff --git a/finetrainers/ltx_video/ltx_video_lora.py b/finetrainers/ltx_video/lora.py similarity index 100% rename from finetrainers/ltx_video/ltx_video_lora.py rename to finetrainers/ltx_video/lora.py diff --git a/tests/test_model_runs_minimally_lora.sh b/tests/test_model_runs_minimally_lora.sh old mode 100644 new mode 100755