Model info api: add aura-flow, hunyuan-dit, flux

2024-08-05 00:25:26 +02:00
parent 42965013f9
commit b5fec4a062
1 changed files with 5 additions and 0 deletions
@@ -25,6 +25,11 @@ model_names = {
    "Stable_Cascade_B": "cascade-b",
    "Stable_Cascade_C": "cascade-c",
    "SD3": "sd3",
+    "AuraFlow": "aura-flow",
+    "HunyuanDiT": "hunyuan-dit",
+    "HunyuanDiT1": "hunyuan-dit",
+    "Flux": "flux",
+    "FluxSchnell": "flux-schnell",
 }