Run Ruff

Author: brandonrising

invokeai/app/invocations/flux_text_to_image.py

@@ -144,10 +144,7 @@ def _run_diffusion(
         # if the cache is not empty.
         # context.models._services.model_manager.load.ram_cache.make_room(24 * 2**30)
 
-        with (
-            transformer_info as transformer,
-            scheduler_info as scheduler
-        ):
+        with transformer_info as transformer, scheduler_info as scheduler:
             assert isinstance(transformer, FluxTransformer2DModel)
             assert isinstance(scheduler, FlowMatchEulerDiscreteScheduler)
 

invokeai/app/invocations/model.py

@@ -60,11 +60,11 @@ class CLIPField(BaseModel):
     loras: List[LoRAField] = Field(description="LoRAs to apply on model loading")
 
 
-
 class TransformerField(BaseModel):
     transformer: ModelIdentifierField = Field(description="Info to load Transformer submodel")
     scheduler: ModelIdentifierField = Field(description="Info to load scheduler submodel")
 
+
 class T5EncoderField(BaseModel):
     tokenizer: ModelIdentifierField = Field(description="Info to load tokenizer submodel")
     text_encoder: ModelIdentifierField = Field(description="Info to load text_encoder submodel")

invokeai/backend/model_manager/load/model_util.py

@@ -50,9 +50,7 @@ def calc_model_size_by_data(logger: logging.Logger, model: AnyModel) -> int:
         return model.calc_size()
     elif isinstance(
         model,
-        (
-            T5TokenizerFast,
-        ),
+        (T5TokenizerFast,),
     ):
         return len(model)
     else:

invokeai/backend/model_manager/util/select_hf_files.py

@@ -54,7 +54,7 @@ def filter_files(
                 "lora_weights.safetensors",
                 "weights.pb",
                 "onnx_data",
-                "spiece.model", # Added for `black-forest-labs/FLUX.1-schnell`.
+                "spiece.model",  # Added for `black-forest-labs/FLUX.1-schnell`.
             )
         ):
             paths.append(file)

invokeai/backend/quantization/fast_quantized_diffusion_model.py

@@ -19,7 +19,7 @@
 
 class FastQuantizedDiffusersModel(QuantizedDiffusersModel):
     @classmethod
-    def from_pretrained(cls, model_name_or_path: Union[str, os.PathLike], base_class = FluxTransformer2DModel, **kwargs):
+    def from_pretrained(cls, model_name_or_path: Union[str, os.PathLike], base_class=FluxTransformer2DModel, **kwargs):
         """We override the `from_pretrained()` method in order to use our custom `requantize()` implementation."""
         base_class = base_class or cls.base_class
         if base_class is None:

invokeai/backend/quantization/fast_quantized_transformers_model.py

@@ -15,7 +15,9 @@
 
 class FastQuantizedTransformersModel(QuantizedTransformersModel):
     @classmethod
-    def from_pretrained(cls, model_name_or_path: Union[str, os.PathLike], auto_class = AutoModelForTextEncoding, **kwargs):
+    def from_pretrained(
+        cls, model_name_or_path: Union[str, os.PathLike], auto_class=AutoModelForTextEncoding, **kwargs
+    ):
         """We override the `from_pretrained()` method in order to use our custom `requantize()` implementation."""
         auto_class = auto_class or cls.auto_class
         if auto_class is None: