Add Lumina 2 to known base models

2025-02-13 16:48:23 +01:00
parent facd65995a
commit ea755151fe
2 changed files with 2 additions and 1 deletions
@@ -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`

@@ -38,6 +38,7 @@ model_names = {
    "GenmoMochi": "mochi",
    "LTXV": "ltxv",
    "HunyuanVideo": "hunyuan-video",
+    "Lumina2": "lumina2",
 }

 gguf_architectures = {"sd1": "sd15"}