pavanperi's picture
Add files using upload-large-folder tool
37f197f verified
Raw
History Blame Contribute Delete
553 Bytes
default_stage:
default_modifiers:
AWQModifier:
targets: [Linear]
ignore: [lm_head]
scheme: W4A16
mappings:
- smooth_layer: re:.*input_layernorm$
balance_layers: ['re:.*query_key_value$', 're:.*attention.dense$']
- 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: true
n_grid: 4