Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

cleanup

Browse Source
This commit is contained in:
Disty0
2025-11-25 23:02:01 +03:00
parent c1a7424c76
commit 70b96daa63
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sdnq/dequantizer.py
+1 -2
View File
@@ -110,8 +110,7 @@ def re_quantize_int_mm(weight: torch.FloatTensor) -> Tuple[torch.Tensor, torch.F
if weight.ndim > 2: # convs
weight = weight.flatten(1,-1)
if use_contiguous_mm:
weight, scale = quantize_int_mm(weight.t(), dim=-0)
weight, scale = weight.contiguous(), scale.contiguous()
weight, scale = quantize_int_mm(weight.t().contiguous(), dim=0)
else:
weight, scale = quantize_int_mm(weight.contiguous(), dim=-1)
weight, scale = weight.t_(), scale.t_()