Merge pull request #139 from Haidra-Org/main

feat: flux support

.github/workflows/prtests.yml

@@ -23,6 +23,8 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v3
+        with:
+          ref: ${{ github.event.pull_request.head.sha }}
       - name: Run pre-commit
         uses: pre-commit/action@v3.0.0
         with:

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.
 

horde_model_reference/meta_consts.py

@@ -83,13 +83,14 @@ class MODEL_PURPOSE(StrEnum):
 
 
 class STABLE_DIFFUSION_BASELINE_CATEGORY(StrEnum):
-    """An enum of all the stable diffusion baselines."""
+    """An enum of all the image generation baselines."""
 
     stable_diffusion_1 = auto()
     stable_diffusion_2_768 = auto()
     stable_diffusion_2_512 = auto()
     stable_diffusion_xl = auto()
     stable_cascade = auto()
+    flux_1 = auto()  # TODO: Extract flux and create "IMAGE_GENERATION_BASELINE_CATEGORY" due to name inconsistency
 
 
 MODEL_PURPOSE_LOOKUP: dict[MODEL_REFERENCE_CATEGORY, MODEL_PURPOSE] = {
@@ -110,6 +111,7 @@ class STABLE_DIFFUSION_BASELINE_CATEGORY(StrEnum):
     STABLE_DIFFUSION_BASELINE_CATEGORY.stable_diffusion_2_512: 512,
     STABLE_DIFFUSION_BASELINE_CATEGORY.stable_diffusion_xl: 1024,
     STABLE_DIFFUSION_BASELINE_CATEGORY.stable_cascade: 1024,
+    STABLE_DIFFUSION_BASELINE_CATEGORY.flux_1: 1024,
 }
 """The single-side preferred resolution for each known stable diffusion baseline."""
 

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"