mirror of
https://github.com/storytold/FineTrainers-Conditioning.git
synced 2026-10-09 00:09:45 +00:00
Aryan
9ef58e2f3a
* rename files * ltx finetuning * update * update * improvements * make style * gradient clipping * update * fix distributed inference * update * update
17 lines
457 B
Python
17 lines
457 B
Python
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]
|