From 157c75888798249518cc42ec5739ce7a9465c087 Mon Sep 17 00:00:00 2001
From: db0 <mail@dbzer0.com>
Date: Wed, 11 Sep 2024 16:26:52 +0200
Subject: [PATCH] feat: support for flux

---
 README.md                            | 1 +
 horde_model_reference/meta_consts.py | 1 +
 stable_diffusion.schema.json         | 5 +++--
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index abd1010..fb803c4 100644
--- a/README.md
+++ b/README.md
@@ -24,6 +24,7 @@ Some key takeaways for the new `stable_diffusion.json`:
   - `stable_diffusion_2_512`
   - `stable_diffusion_xl`
   - `stable_cascade`
+  - `flux_1`
 - An MD5 sum is no longer included. All models (of all types) will have an SHA included from now on.
 - `download` entries optionally contain a new key, `known_slow_download`, which indicates this download host is known to be slow at times.
 
diff --git a/horde_model_reference/meta_consts.py b/horde_model_reference/meta_consts.py
index b89dfcb..b94a843 100644
--- a/horde_model_reference/meta_consts.py
+++ b/horde_model_reference/meta_consts.py
@@ -90,6 +90,7 @@ class STABLE_DIFFUSION_BASELINE_CATEGORY(StrEnum):
     stable_diffusion_2_512 = auto()
     stable_diffusion_xl = auto()
     stable_cascade = auto()
+    flux_1 = auto()
 
 
 MODEL_PURPOSE_LOOKUP: dict[MODEL_REFERENCE_CATEGORY, MODEL_PURPOSE] = {
diff --git a/stable_diffusion.schema.json b/stable_diffusion.schema.json
index 5ee10b1..79d120f 100644
--- a/stable_diffusion.schema.json
+++ b/stable_diffusion.schema.json
@@ -74,13 +74,14 @@
             "type": "string"
         },
         "STABLE_DIFFUSION_BASELINE_CATEGORY": {
-            "description": "An enum of all the stable diffusion baselines.",
+            "description": "An enum of all the image generation baselines.",
             "enum": [
                 "stable_diffusion_1",
                 "stable_diffusion_2_768",
                 "stable_diffusion_2_512",
                 "stable_diffusion_xl",
-                "stable_cascade"
+                "stable_cascade",
+                "flux_1"
             ],
             "title": "STABLE_DIFFUSION_BASELINE_CATEGORY",
             "type": "string"