Downloads · 30 days
4
8% of all-time downloads
ss_ablation_pgd_d3072
asher577/ss_ablation_pgd_d3072
ss_ablation_pgd_d3072 is a machine learning model from asher577. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Weight-sparse transformer trained with the procedure from Gao et al. (2025).
All-time downloads
48
Public
Repo size
1.5 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin1 GB · 68%
At a glance
- Access
- Public
- Created
- Mar 19, 2026
- Updated
- Mar 19, 2026
- SHA
- 6e9c523d
Model card
From the Hugging Face model README
ss_ablation_pgd_d3072
Weight-sparse transformer trained with the procedure from Gao et al. (2025).
Model Details
- Layers: 2
- Model Dimension: 3072
- Context Length: 512
- Head Dimension: 16
- Vocabulary Size: 4096
Sparsity
- Weight Sparsity: True
- Target L0 Fraction: 0.01
- Activation Sparsity: False
Training
- Dataset: jacobcd52/simplestories-tokenized
- Tokenizer: SimpleStories/SimpleStories-1.25M
- Total Tokens: 1,500,000,000
Training Run
Usage
import torch
from huggingface_hub import hf_hub_download
# Download model
model_path = hf_hub_download(repo_id="asher577/ss_ablation_pgd_d3072", filename="pytorch_model.bin")
config_path = hf_hub_download(repo_id="asher577/ss_ablation_pgd_d3072", filename="config.json")
# Load (requires the SparseGPT model class from this repo)
state_dict = torch.load(model_path)
- Created
- Mar 19, 2026
- Updated
- Mar 19, 2026