Update utils.py
This commit is contained in:
parent
0eee505371
commit
189146e389
1 changed files with 1 additions and 0 deletions
|
|
@ -452,6 +452,7 @@ pass
|
|||
def matmul_lora(X, W, W_quant, A, B, s, out = None):
|
||||
dtype = X.dtype
|
||||
W = fast_dequantize(W.t(), W_quant, use_global_buffer = False)
|
||||
print(W)
|
||||
if X.device != W.device:
|
||||
print(X.device, W.device, torch.cuda.current_device())
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue