add bria

Author: Cui-yshoho

docs/diffusers/_toctree.yml

@@ -247,6 +247,8 @@
         title: AllegroTransformer3DModel
       - local: api/models/aura_flow_transformer2d
         title: AuraFlowTransformer2DModel
+      - local: api/models/bria_transformer
+        title: BriaTransformer2DModel
       - local: api/models/chroma_transformer
         title: ChromaTransformer2DModel
       - local: api/models/cogvideox_transformer3d
@@ -369,6 +371,8 @@
       title: AutoPipeline
     - local: api/pipelines/blip_diffusion
       title: BLIP-Diffusion
+    - local: api/pipelines/bria_3_2
+      title: Bria 3.2
     - local: api/pipelines/chroma
       title: Chroma
     - local: api/pipelines/cogvideox

docs/diffusers/api/models/bria_transformer.md

@@ -0,0 +1,17 @@
+<!--Copyright 2025 The HuggingFace Team. All rights reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+the License. You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+specific language governing permissions and limitations under the License.
+-->
+
+# BriaTransformer2DModel
+
+A modified flux Transformer model from [Bria](https://huggingface.co/briaai/BRIA-3.2)
+
+::: BriaTransformer2DModel

docs/diffusers/api/pipelines/bria_3_2.md

@@ -0,0 +1,39 @@
+<!--Copyright 2025 The HuggingFace Team. All rights reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+the License. You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+specific language governing permissions and limitations under the License.
+-->
+
+# Bria 3.2
+
+Bria 3.2 is the next-generation commercial-ready text-to-image model. With just 4 billion parameters, it provides exceptional aesthetics and text rendering, evaluated to provide on par results to leading open-source models, and outperforming other licensed models.
+In addition to being built entirely on licensed data, 3.2 provides several advantages for enterprise and commercial use:
+
+- Efficient Compute - the model is X3 smaller than the equivalent models in the market (4B parameters vs 12B parameters other open source models)
+- Architecture Consistency: Same architecture as 3.1—ideal for users looking to upgrade without disruption.
+- Fine-tuning Speedup: 2x faster fine-tuning on L40S and A100.
+
+Original model checkpoints for Bria 3.2 can be found [here](https://huggingface.co/briaai/BRIA-3.2).
+Github repo for Bria 3.2 can be found [here](https://github.com/Bria-AI/BRIA-3.2).
+
+If you want to learn more about the Bria platform, and get free traril access, please visit [bria.ai](https://bria.ai).
+
+
+## Usage
+
+_As the model is gated, before using it with diffusers you first need to go to the [Bria 3.2 Hugging Face page](https://huggingface.co/briaai/BRIA-3.2), fill in the form and accept the gate. Once you are in, you need to login so that your system knows you’ve accepted the gate._
+
+Use the command below to log in:
+
+```bash
+hf auth login
+```
+
+
+::: BriaPipeline

mindone/diffusers/__init__.py

@@ -70,6 +70,7 @@
         "AutoencoderOobleck",
         "AutoencoderTiny",
         "AutoModel",
+        "BriaTransformer2DModel",
         "ChromaTransformer2DModel",
         "CogVideoXTransformer3DModel",
         "CogView3PlusTransformer2DModel",
@@ -160,6 +161,7 @@
         "AutoPipelineForText2Image",
         "BlipDiffusionControlNetPipeline",
         "BlipDiffusionPipeline",
+        "BriaPipeline",
         "ChromaImg2ImgPipeline",
         "ChromaPipeline",
         "CLIPImageProjection",
@@ -442,6 +444,7 @@
         AutoencoderOobleck,
         AutoencoderTiny,
         AutoModel,
+        BriaTransformer2DModel,
         ChromaTransformer2DModel,
         CogVideoXTransformer3DModel,
         CogView3PlusTransformer2DModel,
@@ -544,6 +547,7 @@
         AutoPipelineForText2Image,
         BlipDiffusionControlNetPipeline,
         BlipDiffusionPipeline,
+        BriaPipeline,
         ChromaImg2ImgPipeline,
         ChromaPipeline,
         CLIPImageProjection,

mindone/diffusers/hooks/_helpers.py

@@ -144,6 +144,7 @@ def _register_attention_processors_metadata():
 def _register_transformer_blocks_metadata():
     from ..models.attention import BasicTransformerBlock
     from ..models.transformers.cogvideox_transformer_3d import CogVideoXBlock
+    from ..models.transformers.transformer_bria import BriaTransformerBlock
     from ..models.transformers.transformer_cogview4 import CogView4TransformerBlock
     from ..models.transformers.transformer_flux import FluxSingleTransformerBlock, FluxTransformerBlock
     from ..models.transformers.transformer_hunyuan_video import (
@@ -166,6 +167,13 @@ def _register_transformer_blocks_metadata():
             return_encoder_hidden_states_index=None,
         ),
     )
+    TransformerBlockRegistry.register(
+        model_class=BriaTransformerBlock,
+        metadata=TransformerBlockMetadata(
+            return_hidden_states_index=0,
+            return_encoder_hidden_states_index=None,
+        ),
+    )
 
     # CogVideoX
     TransformerBlockRegistry.register(

mindone/diffusers/models/__init__.py

@@ -69,6 +69,7 @@
     "transformers.t5_film_transformer": ["T5FilmDecoder"],
     "transformers.transformer_2d": ["Transformer2DModel"],
     "transformers.transformer_allegro": ["AllegroTransformer3DModel"],
+    "transformers.transformer_bria": ["BriaTransformer2DModel"],
     "transformers.transformer_chroma": ["ChromaTransformer2DModel"],
     "transformers.transformer_cogview3plus": ["CogView3PlusTransformer2DModel"],
     "transformers.transformer_cogview4": ["CogView4Transformer2DModel"],
@@ -141,6 +142,7 @@
     from .transformers import (
         AllegroTransformer3DModel,
         AuraFlowTransformer2DModel,
+        BriaTransformer2DModel,
         ChromaTransformer2DModel,
         CogVideoXTransformer3DModel,
         CogView3PlusTransformer2DModel,

mindone/diffusers/models/transformers/__init__.py

@@ -15,6 +15,7 @@
 from .t5_film_transformer import T5FilmDecoder
 from .transformer_2d import Transformer2DModel
 from .transformer_allegro import AllegroTransformer3DModel
+from .transformer_bria import BriaTransformer2DModel
 from .transformer_chroma import ChromaTransformer2DModel
 from .transformer_cogview3plus import CogView3PlusTransformer2DModel
 from .transformer_cogview4 import CogView4Transformer2DModel