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:
Aryan
2024-12-19 04:27:36 +05:30
committed by GitHub
parent 80d1150a0e
commit 9ef58e2f3a
36 changed files with 3089 additions and 154 deletions
+16
View File
@@ -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]