fix noop placeholder in gptq vendor code

1b5d · Apr 23, 2023 · 5bb2199 · 5bb2199
1 parent 80b944a
commit 5bb2199
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/app/llms/gptq_llama/gptq_llama.py b/app/llms/gptq_llama/gptq_llama.py
@@ -106,7 +106,7 @@ def _load_quant(
     ):  # pylint: disable=too-many-arguments
         config = LlamaConfig.from_pretrained(model)
 
-        def noop():
+        def noop(*args, **kwargs):  # pylint: disable=unused-argument
             pass
 
         torch.nn.init.kaiming_uniform_ = noop