From 162e6cd50dc745656d4685b3159c7446f4fa7dc5 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Thu, 9 Jan 2025 13:25:01 +0530 Subject: [PATCH] fix subfolder bug. --- finetrainers/cogvideox/cogvideox_lora.py | 2 +- finetrainers/hunyuan_video/hunyuan_video_lora.py | 2 +- finetrainers/ltx_video/ltx_video_lora.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/finetrainers/cogvideox/cogvideox_lora.py b/finetrainers/cogvideox/cogvideox_lora.py index 9d3a77c..c1413ae 100644 --- a/finetrainers/cogvideox/cogvideox_lora.py +++ b/finetrainers/cogvideox/cogvideox_lora.py @@ -42,7 +42,7 @@ def load_diffusion_models( cache_dir: Optional[str] = None, **kwargs, ): - subfolder = kwargs.get("subfolder", None) + subfolder = kwargs.get("subfolder", "transformer") transformer = CogVideoXTransformer3DModel.from_pretrained( model_id, subfolder=subfolder, torch_dtype=transformer_dtype, revision=revision, cache_dir=cache_dir ) diff --git a/finetrainers/hunyuan_video/hunyuan_video_lora.py b/finetrainers/hunyuan_video/hunyuan_video_lora.py index 450101d..c23feb8 100644 --- a/finetrainers/hunyuan_video/hunyuan_video_lora.py +++ b/finetrainers/hunyuan_video/hunyuan_video_lora.py @@ -63,7 +63,7 @@ def load_diffusion_models( cache_dir: Optional[str] = None, **kwargs, ) -> Dict[str, Union[nn.Module, FlowMatchEulerDiscreteScheduler]]: - subfolder = kwargs.get("subfolder", None) + subfolder = kwargs.get("subfolder", "transformer") transformer = HunyuanVideoTransformer3DModel.from_pretrained( model_id, subfolder=subfolder, torch_dtype=transformer_dtype, revision=revision, cache_dir=cache_dir ) diff --git a/finetrainers/ltx_video/ltx_video_lora.py b/finetrainers/ltx_video/ltx_video_lora.py index 7a80543..a6e8222 100644 --- a/finetrainers/ltx_video/ltx_video_lora.py +++ b/finetrainers/ltx_video/ltx_video_lora.py @@ -45,7 +45,7 @@ def load_diffusion_models( cache_dir: Optional[str] = None, **kwargs, ) -> Dict[str, nn.Module]: - subfolder = kwargs.get("subfolder", None) + subfolder = kwargs.get("subfolder", "transformer") transformer = LTXVideoTransformer3DModel.from_pretrained( model_id, subfolder=subfolder, torch_dtype=transformer_dtype, revision=revision, cache_dir=cache_dir )