diff --git a/README.md b/README.md index 4e244d1..0b589dc 100644 --- a/README.md +++ b/README.md @@ -177,7 +177,7 @@ Lists available models with additional classification info: ... } ``` -Possible values for base model: `sd15, sd20, sd21, sd3, sdxl, sdxl-refiner, ssd1b, svd, cascade-b, cascade-c, aura-flow, hunyuan-dit, flux, flux-schnell` +Possible values for base model: `sd15, sd20, sd21, sd3, sdxl, sdxl-refiner, ssd1b, svd, cascade-b, cascade-c, aura-flow, hunyuan-dit, flux, flux-schnell, lumina2` If base model is `sdxl`, the `type` attribute is set with possible values: `eps, edm, v-prediction, v-prediction-edm` diff --git a/api.py b/api.py index c6ab03c..e9216e7 100644 --- a/api.py +++ b/api.py @@ -38,6 +38,7 @@ model_names = { "GenmoMochi": "mochi", "LTXV": "ltxv", "HunyuanVideo": "hunyuan-video", + "Lumina2": "lumina2", } gguf_architectures = {"sd1": "sd15"}