Skip to content

Commit e516f0b

Browse files
committed
correct import path
1 parent 8e787b6 commit e516f0b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/quantization/test_mixed_precision.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
import torch.nn as nn
55
from torchao.quantization import quantize_, int8_weight_only, int4_weight_only
66
from torchao.quantization.utils import compute_error
7-
from torchao.quantization.prototype.mixed_precision.naive_intNwo import intN_weight_only
7+
from torchao.quantization.prototype.mixed_precision.scripts.naive_intNwo import intN_weight_only
88

99
_CUDA_IS_AVAILABLE = torch.cuda.is_available()
1010

0 commit comments

Comments
 (0)