mirror of
https://github.com/storytold/FineTrainers-Conditioning.git
synced 2026-10-09 00:09:45 +00:00
ArEnSc
e5df80cc36
* Full Finetuning for LTX possibily extended to other models. * Change name of the flag * Used disable grad for component on lora fine tuning enabled * Suggestions Addressed Renamed to SFT Added 2 other models. Testing required. * Switching to Full FineTuning * Run linter. * parse subfolder when needed. * tackle saving and loading hooks. * tackle validation. * fix subfolder bug. * remove __class__. * refactor * remove unnecessary changes * handle saving of final model weights correctly * remove unnecessary changes * LTX uses a default frame rate of 24 FPS We need to modify the output validation framerate to match that value. Add Framerate args. Add Update video output and inference frame rate * There was a results_args mapping that needed to be modified. * update * update README * Update README.md * update docs * add training configuration in cogvideox --------- Co-authored-by: Sayak Paul <spsayakpaul@gmail.com> Co-authored-by: Aryan <aryan@huggingface.co> Co-authored-by: Aryan <contact.aryanvs@gmail.com>
31 lines
897 B
Python
31 lines
897 B
Python
from diffusers import HunyuanVideoPipeline
|
|
|
|
from .hunyuan_video_lora import (
|
|
collate_fn_t2v,
|
|
forward_pass,
|
|
initialize_pipeline,
|
|
load_condition_models,
|
|
load_diffusion_models,
|
|
load_latent_models,
|
|
post_latent_preparation,
|
|
prepare_conditions,
|
|
prepare_latents,
|
|
validation,
|
|
)
|
|
|
|
|
|
# TODO(aryan): refactor into model specs for better re-use
|
|
HUNYUAN_VIDEO_T2V_FULL_FINETUNE_CONFIG = {
|
|
"pipeline_cls": HunyuanVideoPipeline,
|
|
"load_condition_models": load_condition_models,
|
|
"load_latent_models": load_latent_models,
|
|
"load_diffusion_models": load_diffusion_models,
|
|
"initialize_pipeline": initialize_pipeline,
|
|
"prepare_conditions": prepare_conditions,
|
|
"prepare_latents": prepare_latents,
|
|
"post_latent_preparation": post_latent_preparation,
|
|
"collate_fn": collate_fn_t2v,
|
|
"forward_pass": forward_pass,
|
|
"validation": validation,
|
|
}
|