quantization_config.json

{
  "bits": 4,
  "group_size": 128,
  "sym": true,
  "data_type": "int",
  "enable_quanted_input": true,
  "enable_minmax_tuning": true,
  "seqlen": 2048,
  "batch_size": 8,
  "scale_dtype": "torch.float16",
  "lr": 0.001,
  "minmax_lr": 0.001,
  "gradient_accumulate_steps": 1,
  "iters": 1000,
  "amp": true,
  "nsamples": 512,
  "low_gpu_mem_usage": false,
  "quant_block_list": null,
  "enable_norm_bias_tuning": false,
  "dataset": "NeelNanda/pile-10k",
  "autoround_version": "0.4.0.dev",
  "quant_method": "auto-round",
  "extra_config": {
    "model.layers.5.mlp.gate_proj": {
      "data_type": "float",
      "bits": 16
    },
    "model.layers.5.mlp.up_proj": {
      "data_type": "float",
      "bits": 16
    },
    "model.layers.5.mlp.down_proj": {
      "data_type": "float",
      "bits": 16
    }
  }
}