From 9ef7b913e2b4089ec9c2aba4b6c654f9ddfd3810 Mon Sep 17 00:00:00 2001
From: ciaranbor <ciaranborourke-dev@proton.me>
Date: Fri, 28 Nov 2025 11:53:49 +0000
Subject: [PATCH] ADd flux1-schnell model

---
 src/exo/shared/models/model_cards.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/exo/shared/models/model_cards.py b/src/exo/shared/models/model_cards.py
index 6ef9e666..ae677fb3 100644
--- a/src/exo/shared/models/model_cards.py
+++ b/src/exo/shared/models/model_cards.py
@@ -601,4 +601,18 @@ MODEL_CARDS: dict[str, ModelCard] = {
     #         supports_tensor=True,
     #     ),
     # ),
+    "flux1-schnell": ModelCard(
+        short_id="flux1-schnell",
+        model_id="black-forest-labs/FLUX.1-schnell",
+        name="FLUX.1 [schnell]",
+        description="""FLUX.1 [schnell] is a 12 billion parameter rectified flow transformer capable of generating images from text descriptions""",
+        task="text-generation",
+        tags=[],
+        metadata=ModelMetadata(
+            model_id=ModelId("black-forest-labs/FLUX.1-schnell"),
+            pretty_name="FLUX.1 [schnell]",
+            storage_size=Memory.from_kb(327512),  # TODO
+            n_layers=28,  # TODO
+        ),
+    ),
 }