feat: support for flux

Haidra-Org · Sep 11, 2024 · 157c758 · 157c758
1 parent 4be901a
commit 157c758
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git 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
@@ -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
@@ -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"