style fixs

Author: galbria

src/diffusers/__init__.py

@@ -198,8 +198,8 @@
             "AutoencoderOobleck",
             "AutoencoderTiny",
             "AutoModel",
-            "BriaTransformer2DModel",
             "BriaFiboTransformer2DModel",
+            "BriaTransformer2DModel",
             "CacheMixin",
             "ChromaTransformer2DModel",
             "CogVideoXTransformer3DModel",
@@ -393,8 +393,8 @@
 else:
     _import_structure["modular_pipelines"].extend(
         [
-            "BriaFiboVLMPromptToJson",
             "BriaFiboGeminiPromptToJson",
+            "BriaFiboVLMPromptToJson",
             "FluxAutoBlocks",
             "FluxKontextAutoBlocks",
             "FluxKontextModularPipeline",
@@ -433,8 +433,8 @@
             "AuraFlowPipeline",
             "BlipDiffusionControlNetPipeline",
             "BlipDiffusionPipeline",
-            "BriaPipeline",
             "BriaFiboPipeline",
+            "BriaPipeline",
             "ChromaImg2ImgPipeline",
             "ChromaPipeline",
             "CLIPImageProjection",
@@ -905,8 +905,8 @@
             AutoencoderOobleck,
             AutoencoderTiny,
             AutoModel,
-            BriaTransformer2DModel,
             BriaFiboTransformer2DModel,
+            BriaTransformer2DModel,
             CacheMixin,
             ChromaTransformer2DModel,
             CogVideoXTransformer3DModel,
@@ -1108,8 +1108,8 @@
             AudioLDM2UNet2DConditionModel,
             AudioLDMPipeline,
             AuraFlowPipeline,
-            BriaPipeline,
             BriaFiboPipeline,
+            BriaPipeline,
             ChromaImg2ImgPipeline,
             ChromaPipeline,
             CLIPImageProjection,

src/diffusers/models/__init__.py

@@ -175,8 +175,8 @@
         from .transformers import (
             AllegroTransformer3DModel,
             AuraFlowTransformer2DModel,
-            BriaTransformer2DModel,
             BriaFiboTransformer2DModel,
+            BriaTransformer2DModel,
             ChromaTransformer2DModel,
             CogVideoXTransformer3DModel,
             CogView3PlusTransformer2DModel,

src/diffusers/modular_pipelines/bria_fibo/__init__.py

@@ -21,8 +21,8 @@
 
     _dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
 else:
-    _import_structure["gemini_prompt_to_json"] = ["BriaFiboGeminiPromptToJson"]
     _import_structure["fibo_vlm_prompt_to_json"] = ["BriaFiboVLMPromptToJson"]
+    _import_structure["gemini_prompt_to_json"] = ["BriaFiboGeminiPromptToJson"]
 
 if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
     try:
@@ -31,8 +31,8 @@
     except OptionalDependencyNotAvailable:
         from ...utils.dummy_torch_and_transformers_objects import *  # noqa F403
     else:
-        from .gemini_prompt_to_json import BriaFiboGeminiPromptToJson
         from .fibo_vlm_prompt_to_json import BriaFiboVLMPromptToJson
+        from .gemini_prompt_to_json import BriaFiboGeminiPromptToJson
 else:
     import sys
 

src/diffusers/modular_pipelines/bria_fibo/fibo_vlm_prompt_to_json.py

@@ -277,16 +277,12 @@ def build_messages(
         if refine_image is None:
             base_prompt = (structured_prompt or "").strip()
             edits = (editing_instructions or "").strip()
-            formatted = textwrap.dedent(
-                f"""<refine> Input: {base_prompt} Editing instructions: {edits}"""
-            ).strip()
+            formatted = textwrap.dedent(f"""<refine> Input: {base_prompt} Editing instructions: {edits}""").strip()
             user_content.append({"type": "text", "text": formatted})
         else:
             user_content.append({"type": "image", "image": refine_image})
             edits = (editing_instructions or "").strip()
-            formatted = textwrap.dedent(
-                f"""<refine> Editing instructions: {edits}"""
-            ).strip()
+            formatted = textwrap.dedent(f"""<refine> Editing instructions: {edits}""").strip()
             user_content.append({"type": "text", "text": formatted})
 
     messages: List[Dict[str, Any]] = []

tests/models/transformers/test_models_transformer_bria_fibo.py

@@ -18,15 +18,14 @@
 import torch
 
 from diffusers import BriaFiboTransformer2DModel
+
 from ...testing_utils import enable_full_determinism, torch_device
 from ..test_modeling_common import LoraHotSwappingForModelTesterMixin, ModelTesterMixin, TorchCompileTesterMixin
 
 
 enable_full_determinism()
 
 
-
-
 class BriaFiboTransformerTests(ModelTesterMixin, unittest.TestCase):
     model_class = BriaFiboTransformer2DModel
     main_input_name = "hidden_states"
@@ -57,7 +56,7 @@ def dummy_input(self):
             "img_ids": image_ids,
             "txt_ids": text_ids,
             "timestep": timestep,
-            "text_encoder_layers": [encoder_hidden_states[:,:,:32], encoder_hidden_states[:,:,:32]],
+            "text_encoder_layers": [encoder_hidden_states[:, :, :32], encoder_hidden_states[:, :, :32]],
         }
 
     @property

tests/pipelines/bria_fibo/test_pipeline_bria_fibo.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import gc
 import tempfile
 import unittest
 
@@ -30,10 +29,8 @@
 from tests.pipelines.test_pipelines_common import PipelineTesterMixin, to_np
 
 from ...testing_utils import (
-    backend_empty_cache,
     enable_full_determinism,
     require_torch_accelerator,
-    slow,
     torch_device,
 )
 
@@ -194,5 +191,3 @@ def test_to_dtype(self):
 
     def test_save_load_dduf(self):
         pass
-
-