We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8f8559b commit ce64ef8Copy full SHA for ce64ef8
neural_compressor/torch/quantization/autotune.py
@@ -29,7 +29,7 @@
29
30
def get_default_tune_config() -> TuningConfig:
31
# TODO use the registered default tuning config in the next PR
32
- return TuningConfig(quant_configs=[GPTQConfig(weight_bits=[4, 8]), RTNConfig(weight_bits=[4, 8])])
+ return TuningConfig(quant_configs=[GPTQConfig(weight_bits=[4, 8]), RTNConfig(bits=[4, 8])])
33
34
35
def autotune(
0 commit comments