18 lines
660 B
YAML
18 lines
660 B
YAML
default_stage:
|
|
default_modifiers:
|
|
AWQModifier:
|
|
targets: [Linear]
|
|
ignore: ['re:.*lm_head.*', 're:.*vision.*', 're:.*visual.*']
|
|
scheme: W4A16_ASYM
|
|
mappings:
|
|
- smooth_layer: re:.*input_layernorm$
|
|
balance_layers: ['re:.*q_proj$', 're:.*k_proj$', 're:.*v_proj$']
|
|
- smooth_layer: re:.*v_proj$
|
|
balance_layers: ['re:.*o_proj$']
|
|
- smooth_layer: re:.*post_attention_layernorm$
|
|
balance_layers: ['re:.*gate_proj$', 're:.*up_proj$']
|
|
- smooth_layer: re:.*up_proj$
|
|
balance_layers: ['re:.*down_proj$']
|
|
offload_device: !!python/object/apply:torch.device [cpu]
|
|
duo_scaling: false
|