lint

Signed-off-by: Andrea Fasoli <[email protected]>

Author: andrea-fasoli

fms_mo/aiu_addons/i8i8/i8i8_aiu_adapter.py

@@ -98,7 +98,13 @@ def _add_defaults_and_concat(
 
 
 # registration of new adapter step and adapter for each architecture
-for arch in ["llama", "gpt_bigcode", "granite", "roberta", "roberta_question_answering"]:
+for arch in [
+    "llama",
+    "gpt_bigcode",
+    "granite",
+    "roberta",
+    "roberta_question_answering",
+]:
     serialization.register_adapter_step(arch, "int8_qparams_aiu", _int8_qparams_aiu)
     if arch in ["llama", "granite"]:
         steps_to_register = [

fms_mo/aiu_addons/i8i8/i8i8_aiu_linear.py

@@ -87,9 +87,7 @@ def __init__(
 
         self.has_bias = bias
         bias_size = out_features if self.has_bias else 1
-        self.register_buffer(
-            "bias", torch.zeros((bias_size), dtype=torch.float16)
-        )
+        self.register_buffer("bias", torch.zeros((bias_size), dtype=torch.float16))
 
         if config.weight_per_channel:
             w_clip_size = out_features