Merge branch 'main' into grad-update-fixes

a-r-r-o-w · Jan 13, 2025 · 8af6b26 · 8af6b26
2 parents bd9a5a5 + 9f0fdb5
commit 8af6b26
Show file tree

Hide file tree

Showing 10 changed files with 6 additions and 6 deletions.
diff --git 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
@@ -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 → finetrainers/cogvideox/lora.py b/finetrainers/cogvideox/cogvideox_lora.py → finetrainers/cogvideox/lora.py
diff --git 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
@@ -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/...iners/hunyuan_video/hunyuan_video_lora.py → finetrainers/hunyuan_video/lora.py b/...iners/hunyuan_video/hunyuan_video_lora.py → finetrainers/hunyuan_video/lora.py
diff --git 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
@@ -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 → finetrainers/ltx_video/lora.py b/finetrainers/ltx_video/ltx_video_lora.py → finetrainers/ltx_video/lora.py
diff --git a/tests/test_model_runs_minimally_lora.sh b/tests/test_model_runs_minimally_lora.sh