storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-20 07:45:06 +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/models.py
+6 -4
View File
@@ -1,20 +1,22 @@
from typing import Any, Dict
from .cogvideox import COGVIDEOX_T2V_LORA_CONFIG
from .hunyuan_video import HUNYUAN_VIDEO_T2V_LORA_CONFIG
from .ltx_video import LTX_VIDEO_T2V_LORA_CONFIG, LTX_VIDEO_T2V_FT_CONFIG
from .cogvideox import COGVIDEOX_T2V_LORA_CONFIG, COGVIDEOX_T2V_SFT_CONFIG
from .hunyuan_video import HUNYUAN_VIDEO_T2V_LORA_CONFIG, HUNYUAN_VIDEO_T2V_SFT_CONFIG
from .ltx_video import LTX_VIDEO_T2V_LORA_CONFIG, LTX_VIDEO_T2V_SFT_CONFIG
SUPPORTED_MODEL_CONFIGS = {
"hunyuan_video": {
"lora": HUNYUAN_VIDEO_T2V_LORA_CONFIG,
"sft": HUNYUAN_VIDEO_T2V_SFT_CONFIG,
},
"ltx_video": {
"lora": LTX_VIDEO_T2V_LORA_CONFIG,
"finetune": LTX_VIDEO_T2V_FT_CONFIG,
"sft": LTX_VIDEO_T2V_SFT_CONFIG,
},
"cogvideox": {
"lora": COGVIDEOX_T2V_LORA_CONFIG,
"sft": COGVIDEOX_T2V_SFT_CONFIG,
},
}