CoolFace
Modelpublic

LiamCarter/grasprune_qwen3-8b_ratio0.2

sourceHugging Faceapache-2.0updated 22d agoView on Hugging Face
0likes
Model Card

Qwen3-8B — original GRASPrune, pruning ratio 0.2

This is a structurally pruned derivative of `Qwen/Qwen3-8B`.

Model provenance

This checkpoint was produced by the original GRASPrune pruning pipeline in `methods/GRASPrune`, not by `new_method_2`. The local new_method_2 code was used only for downstream evaluation and did not produce or alter the pruned weights.

The 20% ratio is the target reduction of GRASPrune's global prunable structural budget across FFN intermediate channels and attention KV groups. It is not a uniform per-layer sparsity ratio or a file-size reduction ratio.

Files and loading

This is not a standard Transformers save_pretrained directory and cannot be loaded directly with AutoModelForCausalLM.from_pretrained().

  • —pruned_state_dict.safetensors: materialized pruned weights
  • —meta.json: layer-specific shapes required to rebuild the architecture
  • —layer_mask_report.csv: layer-wise retention report
  • —provenance.json: release provenance and artifact identity
  • —SHA256SUMS: checksum for the weight file

Clone the original GRASPrune repository, run from its root, and use its rebuild.py loader:

python
import os
import torch
from huggingface_hub import snapshot_download
from rebuild import load_pruned_model

checkpoint_dir = snapshot_download("LiamCarter/grasprune_qwen3-8b_ratio0.2")
model, tokenizer, meta = load_pruned_model(
    model_id="Qwen/Qwen3-8B",
    state_dict_path=os.path.join(checkpoint_dir, "pruned_state_dict.safetensors"),
    meta_path=os.path.join(checkpoint_dir, "meta.json"),
    torch_dtype=torch.bfloat16,
    device="cuda:0",
    use_fast_tokenizer=True,
    trust_remote_code=True,
    local_only=False,
)

Local evaluation

The weights were evaluated read-only with the local new_method_2 evaluation flow. Percentages:

Common-sense 5-task macro avgICL 0-shotICL 1-shotICL 4-shotICL 8-shot
62.51430.6429.7451.0655.82

These are local benchmark results, not upstream GRASPrune claims.

Intended use and limitations

This checkpoint is intended for structured-pruning and benchmark research. It has not been validated for production deployment, safety, factual reliability, bias, multilingual robustness, or long-context behavior. Performance can be lower than the parent model.

This derivative follows the Apache-2.0 license of the Qwen3-8B base model. GRASPrune code is separately distributed under Apache-2.0.