Update plugins/accelerated-peft/src/fms_acceleration_peft/framework_p…

…lugin_autogptq.py Co-authored-by: Yu Chin Fabian Lim <[email protected]>
foundation-model-stack · Jul 16, 2024 · 354fe0d · 354fe0d
1 parent 0d3ac62
commit 354fe0d
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/plugins/accelerated-peft/src/fms_acceleration_peft/framework_plugin_autogptq.py b/plugins/accelerated-peft/src/fms_acceleration_peft/framework_plugin_autogptq.py
@@ -57,9 +57,11 @@ def __init__(self, configurations: Dict[str, Dict]):
             from transformers.utils.import_utils import _is_package_available # pylint: disable=import-outside-toplevel
             assert (
                 _is_package_available("auto_gptq") is True
-            ), "Unable to use external library, auto_gptq module not found. \
-                Refer to README for installation instructions \
-                as a specific version might be required."
+            ),  (
+               "Unable to use external library, auto_gptq module not found. "
+                "Refer to README for installation instructions  "
+                "as a specific version might be required."
+            )
 
     def model_loader(self, model_name: str, **kwargs):
         # guarded imports