Unverified Commit 76edff62 authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

FIX: 'awq_ext' is not defined error (#465)

parent 4fc6cc03
......@@ -63,7 +63,7 @@ class WQLinearMMFunction(Function):
def backward(ctx, grad_output):
input, qweight, qzeros, scales, bias = ctx.saved_tensors
if awq_ext is None:
if not AWQ_INSTALLED:
raise ValueError(
"auto-awq kernels is needed to be installed to use `.backward()`. Make sure to install the auto-awq kernels"
" by following the installation guides in https://github.com/casper-hansen/AutoAWQ_kernels"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment