From 42c89b8987e47bf96df52b24ff4329ec179e585f Mon Sep 17 00:00:00 2001 From: Kyle Sayers Date: Wed, 19 Feb 2025 18:39:38 -0500 Subject: [PATCH] Fix Readme Imports (#1165) ## Purpose ## * Readme clarity ## Changes ## * Remove unneeded imports in the readme * Change order of imports/appearance to match order of usage Signed-off-by: Kyle Sayers Co-authored-by: Dipika Sikka --- README.md | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/README.md b/README.md index 9021c6193..ffc651da8 100644 --- a/README.md +++ b/README.md @@ -56,10 +56,9 @@ Note that the model can be swapped for a local or remote HF-compatible checkpoin Quantization is applied by selecting an algorithm and calling the `oneshot` API. ```python -from llmcompressor.modifiers.quantization import GPTQModifier from llmcompressor.modifiers.smoothquant import SmoothQuantModifier +from llmcompressor.modifiers.quantization import GPTQModifier from llmcompressor.transformers import oneshot -from transformers import AutoModelForCausalLM # Select quantization algorithm. In this case, we: # * apply SmoothQuant to make the activations easier to quantize