tesseraq_w3g128_L31_8b.yml

base:
    seed: &seed 42
model:
    type: Llama
    path: meta-llama/Meta-Llama-3.1-8B
    torch_dtype: auto
calib:
    name: c4
    download: False
    n_samples: 512
    path: ../cache/data/calib/c4
    bs: 1
    seq_len: 2048
    preproc: c4_gptq
    seed: *seed
eval:
    eval_pos: [fake_quant]
    name: [wikitext2, c4]
    download: False
    path: ../cache/data/eval
    bs: 1
    seq_len: 2048
    inference_per_block: False
    tasks: piqa,arc_easy,arc_challenge,hellaswag,winogrande
quant:
    method: TesseraQ
    weight:
        bit: 3
        symmetric: False
        granularity: per_group
        group_size: 128
        calib_algo: minmax
    special:
        lr: 0.001
        iterations: 250
        wd: 0.0
        batch_size: 4
        deactive_amp: False
        aug_loss: False
        optimize_scale: True
        scale_lr: 0.001
        thresholds: [0.8, 0.65, 0.5, 0.43, 0.38, 0.34, 0.3, 0.27, 0.24, 0.21, 0.18, 0.15, 0.12, 0.10, 0.08,
                      0.06, 0.04, 0.02, 0.01, 0.005]
        weight_clip: True
        load_transform: True
        clip_version: v1
        reduce_memory: False
        scale_path: ../cache/activations/L31_8b/awq_w3g128
        clip_path: ../cache/activations/L31_8b/awq_w3g128
    quant_out: True
save:
    save_fp: False
    save_trans: False
    save_lightllm: False
    save_autogptq: False