ADd flux1-schnell model

2026-01-06 10:08:11 +00:00
parent 0daa4b36db
commit 9ef7b913e2
1 changed files with 14 additions and 0 deletions
@@ -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
+        ),
+    ),
 }