There was an error while loading. Please reload this page.
1 parent e87e780 commit 039338fCopy full SHA for 039338f
1 file changed
nunchaku/ops/gemm.py
@@ -2,6 +2,8 @@
2
Python wrappers for Nunchaku's quantized GEMM operations.
3
"""
4
5
+import math
6
+
7
import torch
8
9
from .._C import ops
@@ -121,7 +123,7 @@ def svdq_gemm_w4a4_cuda(
121
123
122
124
if lora_scales is None:
125
rank = lora_up.shape[1]
- lora_scales = [1.0] * rank
126
+ lora_scales = [1.0] * math.ceil(rank / 16)
127
128
ops.gemm_w4a4(
129
act,
0 commit comments