From b5fec4a0625e47586b7b28585ce27d4ecdd3b18a Mon Sep 17 00:00:00 2001
From: Acly <aclysia@gmail.com>
Date: Mon, 5 Aug 2024 00:25:26 +0200
Subject: [PATCH] Model info api: add aura-flow, hunyuan-dit, flux

---
 api.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/api.py b/api.py
index 7678427..f255436 100644
--- a/api.py
+++ b/api.py
@@ -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",
 }