mirror of
https://github.com/storytold/FineTrainers-Conditioning.git
synced 2026-10-18 00:33:52 +00:00
LTX Video (#123)
* rename files * ltx finetuning * update * update * improvements * make style * gradient clipping * update * fix distributed inference * update * update
This commit is contained in:
@@ -0,0 +1,16 @@
|
||||
from typing import Any, Dict
|
||||
|
||||
from .ltx_video import LTX_VIDEO_T2V_CONFIG
|
||||
|
||||
|
||||
SUPPORTED_MODEL_CONFIGS = {
|
||||
"ltx_video": LTX_VIDEO_T2V_CONFIG,
|
||||
}
|
||||
|
||||
|
||||
def get_config_from_model_name(model_name: str) -> Dict[str, Any]:
|
||||
if model_name not in SUPPORTED_MODEL_CONFIGS:
|
||||
raise ValueError(
|
||||
f"Model {model_name} not supported. Supported models are: {list(SUPPORTED_MODEL_CONFIGS.keys())}"
|
||||
)
|
||||
return SUPPORTED_MODEL_CONFIGS[model_name]
|
||||
Reference in New Issue
Block a user