fmt

Signed-off-by: Thara Palanivel <[email protected]>

Author: tharapalanivel

fms_mo/aiu_addons/gptq/gptq_aiu_adapter.py

@@ -41,7 +41,9 @@ def _gptq_qweights_transpose_aiu(
 serialization.register_adapter_step(
     "gpt_bigcode", "gptq_qweights_transpose_aiu", _gptq_qweights_transpose_aiu
 )
-serialization.register_adapter_step("granite", "gptq_qweights_transpose_aiu", _gptq_qweights_transpose_aiu)
+serialization.register_adapter_step(
+    "granite", "gptq_qweights_transpose_aiu", _gptq_qweights_transpose_aiu
+)
 serialization.register_adapter(
     "llama",
     "hf_gptq_aiu",
@@ -61,6 +63,11 @@ def _gptq_qweights_transpose_aiu(
 serialization.register_adapter(
     "granite",
     "hf_gptq_aiu",
-    ["hf_to_fms_names", "hf_to_fms_rope", "hf_gptq_fusion_check", "weight_fusion", "gptq_qweights_transpose_aiu"]
+    [
+        "hf_to_fms_names",
+        "hf_to_fms_rope",
+        "hf_gptq_fusion_check",
+        "weight_fusion",
+        "gptq_qweights_transpose_aiu",
+    ],
 )
-