From 354fe0da909c35eba63d93c2b7b883cf4a5b02f1 Mon Sep 17 00:00:00 2001 From: achew010 <165894159+achew010@users.noreply.github.com> Date: Tue, 16 Jul 2024 12:30:26 +0800 Subject: [PATCH] Update plugins/accelerated-peft/src/fms_acceleration_peft/framework_plugin_autogptq.py Co-authored-by: Yu Chin Fabian Lim --- .../fms_acceleration_peft/framework_plugin_autogptq.py | 8 +++++--- 1 file changed, 5 insertions(+), 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 index 9ae42ac2..4a1a233d 100644 --- 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