From 6772ffa60201428723fd28aaafdad1d5da104c10 Mon Sep 17 00:00:00 2001
From: Demian Frister <demian.frister@kit.edu>
Date: Tue, 20 Aug 2024 14:04:17 +0000
Subject: [PATCH] flux added

---
 alias-flux.yaml | 14 ++++++++++++++
 index.yaml      | 39 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 53 insertions(+)
 create mode 100644 alias-flux.yaml

diff --git a/alias-flux.yaml b/alias-flux.yaml
new file mode 100644
index 0000000..9c2d428
--- /dev/null
+++ b/alias-flux.yaml
@@ -0,0 +1,14 @@
+---
+name: "flux"
+
+config_file: |
+  backend: diffusers
+  f16: true
+  low_vram: true
+  step: 25
+
+  diffusers:
+    cuda: true
+    enable_parameters: num_inference_steps
+    pipeline_type: FluxPipeline
+    cfg_scale: 0
\ No newline at end of file
diff --git a/index.yaml b/index.yaml
index b507c98..ed7b085 100644
--- a/index.yaml
+++ b/index.yaml
@@ -197,6 +197,45 @@
     - gpu
     - Image generation
     - Diffusion
+
+- &flux
+  name: flux.1-dev
+  license: flux-1-dev-non-commercial-license
+  description: |
+    FLUX.1 [dev] is a 12 billion parameter rectified flow transformer capable of generating images from text descriptions. For more information, please read our blog post.
+    Key Features
+        Cutting-edge output quality, second only to our state-of-the-art model FLUX.1 [pro].
+        Competitive prompt following, matching the performance of closed source alternatives .
+        Trained using guidance distillation, making FLUX.1 [dev] more efficient.
+        Open weights to drive new scientific research, and empower artists to develop innovative workflows.
+        Generated outputs can be used for personal, scientific, and commercial purposes as described in the flux-1-dev-non-commercial-license.
+  urls:
+    - https://huggingface.co/black-forest-labs/FLUX.1-dev
+  tags:
+    - text-to-image
+    - flux
+    - python
+    - gpu
+  url: https://gitlab.kit.edu/kit/aifb/BIS/infrastruktur/localai/localai-model-gallery/-/raw/main/flux.yaml
+  overrides:
+    parameters:
+      model: ChuckMcSneed/FLUX.1-dev
+- !!merge <<: *flux
+  name: flux.1-schnell
+  license: apache-2
+  icon: https://huggingface.co/black-forest-labs/FLUX.1-schnell/resolve/main/schnell_grid.jpeg
+  description: |
+    FLUX.1 [schnell] is a 12 billion parameter rectified flow transformer capable of generating images from text descriptions. For more information, please read our blog post.
+    Key Features
+
+        Cutting-edge output quality and competitive prompt following, matching the performance of closed source alternatives.
+        Trained using latent adversarial diffusion distillation, FLUX.1 [schnell] can generate high-quality images in only 1 to 4 steps.
+        Released under the apache-2.0 licence, the model can be used for personal, scientific, and commercial purposes.
+  urls:
+    - https://huggingface.co/black-forest-labs/FLUX.1-schnell
+  overrides:
+    parameters:
+      model: black-forest-labs/FLUX.1-schnell
     
 # TRANSCRIPTION MODELS
 - url: https://gitlab.kit.edu/kit/aifb/BIS/infrastruktur/localai/localai-model-gallery/-/raw/main/alias-speechtotext.yaml
-- 
GitLab