fix style

huggingface · Nov 27, 2024 · 712d851 · 712d851
1 parent 1b68c85
commit 712d851
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/pytorch_llama.py b/examples/pytorch_llama.py
@@ -33,7 +33,7 @@
         "torch_dtype": "bfloat16",
         "quantization_scheme": "torchao",
         "quantization_config": {"quant_type": "int4_weight_only", "group_size": 128},
-    }
+    },
 }
 
 

diff --git a/optimum_benchmark/backends/pytorch/backend.py b/optimum_benchmark/backends/pytorch/backend.py
@@ -11,12 +11,12 @@
     AwqConfig,
     BitsAndBytesConfig,
     GPTQConfig,
+    TorchAoConfig,
     Trainer,
     TrainerCallback,
     TrainerState,
     TrainingArguments,
 )
-from transformers import TorchAoConfig
 
 from ...import_utils import is_deepspeed_available, is_torch_distributed_available, is_zentorch_available
 from ..base import Backend
-Original file line number
+Diff line change
@@ Expand Up / @@ -33,7 +33,7 @@ @@
             "torch_dtype": "bfloat16",
             "quantization_scheme": "torchao",
             "quantization_config": {"quant_type": "int4_weight_only", "group_size": 128},
-        }
+        },
     }
@@ Expand Down @@