From d9b8f1253d2d6884d894b839ac700bc78e8f1855 Mon Sep 17 00:00:00 2001 From: Qubitium-ModelCloud Date: Fri, 3 Jan 2025 09:50:56 +0800 Subject: [PATCH] update comments --- optimum/gptq/constants.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/optimum/gptq/constants.py b/optimum/gptq/constants.py index 4cff27d0d0..701868a3b8 100644 --- a/optimum/gptq/constants.py +++ b/optimum/gptq/constants.py @@ -18,7 +18,7 @@ "model.decoder.layers", "gpt_neox.layers", "model.layers", - # Models loaded by AutoModel have different block prefixes than models loaded by AutoModelForCausalLM, need to match it + # modules loaded by AutoModel vs AutoModelForCausalLM have different prefixes "h", "decoder.layers", "layers",