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

Suggestions Addressed

Browse Source 
Renamed to SFT
Added 2 other models.
Testing required.
This commit is contained in:
CrossProduct
2025-01-07 22:59:09 +00:00
parent 4cd5a8e53b
commit cb9381b40e
7 changed files with 69 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/cogvideox/cogvideox_lora.py
+15
View File
@@ -325,3 +325,18 @@ COGVIDEOX_T2V_LORA_CONFIG = {
"forward_pass": forward_pass,
"validation": validation,
}
COGVIDEOX_T2V_SFT_CONFIG = {
"pipeline_cls": CogVideoXPipeline,
"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,
"calculate_noisy_latents": calculate_noisy_latents,
"forward_pass": forward_pass,
"validation": validation,
}