storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-17 00:23:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix subfolder bug.

Browse Source
This commit is contained in:
sayakpaul
2025-01-09 13:25:01 +05:30
parent 5183405e61
commit 162e6cd50d
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/cogvideox/cogvideox_lora.py
+1 -1
View File
@@ -42,7 +42,7 @@ def load_diffusion_models(
cache_dir: Optional[str] = None, cache_dir: Optional[str] = None,
**kwargs, **kwargs,
): ):
subfolder = kwargs.get("subfolder", None) subfolder = kwargs.get("subfolder", "transformer")
transformer = CogVideoXTransformer3DModel.from_pretrained( transformer = CogVideoXTransformer3DModel.from_pretrained(
model_id, subfolder=subfolder, torch_dtype=transformer_dtype, revision=revision, cache_dir=cache_dir model_id, subfolder=subfolder, torch_dtype=transformer_dtype, revision=revision, cache_dir=cache_dir
) )
finetrainers/hunyuan_video/hunyuan_video_lora.py
+1 -1
View File
@@ -63,7 +63,7 @@ def load_diffusion_models(
cache_dir: Optional[str] = None, cache_dir: Optional[str] = None,
**kwargs, **kwargs,
) -> Dict[str, Union[nn.Module, FlowMatchEulerDiscreteScheduler]]: ) -> Dict[str, Union[nn.Module, FlowMatchEulerDiscreteScheduler]]:
subfolder = kwargs.get("subfolder", None) subfolder = kwargs.get("subfolder", "transformer")
transformer = HunyuanVideoTransformer3DModel.from_pretrained( transformer = HunyuanVideoTransformer3DModel.from_pretrained(
model_id, subfolder=subfolder, torch_dtype=transformer_dtype, revision=revision, cache_dir=cache_dir model_id, subfolder=subfolder, torch_dtype=transformer_dtype, revision=revision, cache_dir=cache_dir
) )
finetrainers/ltx_video/ltx_video_lora.py
+1 -1
View File
@@ -45,7 +45,7 @@ def load_diffusion_models(
cache_dir: Optional[str] = None, cache_dir: Optional[str] = None,
**kwargs, **kwargs,
) -> Dict[str, nn.Module]: ) -> Dict[str, nn.Module]:
subfolder = kwargs.get("subfolder", None) subfolder = kwargs.get("subfolder", "transformer")
transformer = LTXVideoTransformer3DModel.from_pretrained( transformer = LTXVideoTransformer3DModel.from_pretrained(
model_id, subfolder=subfolder, torch_dtype=transformer_dtype, revision=revision, cache_dir=cache_dir model_id, subfolder=subfolder, torch_dtype=transformer_dtype, revision=revision, cache_dir=cache_dir
) )