README.md

---
license: apache-2.0
tags:
- pruned
- python
- optimized
- wanda
base_model: Qwen/Qwen2.5-0.5B
pipeline_tag: text-generation
---

# Qwen2.5-0.5B-python-safe

> 🎯 **PYTHON-optimized** | 📦 **Safe** pruning | ⚡ **1% weights pruned**

This model is a **conservatively pruned** version of [Qwen/Qwen2.5-0.5B](https://huggingface.co/Qwen/Qwen2.5-0.5B).

## Performance Comparison

| Category | Original | Pruned | Change |
|----------|----------|--------|--------|
| **Python** | 0.0% | 0.0% ⭐ | → |
| Html | 0.0% | 0.0% | → |
| Trivia | 100.0% | 100.0% | → |
| Math | 66.7% | 66.7% | → |
| Reasoning | 66.7% | 66.7% | → |
| Medical | 66.7% | 66.7% | → |
| Linux | 33.3% | 33.3% | → |
| Writing | 33.3% | 33.3% | → |

**Average**: 45.8% → 45.8% (+0.0%)



![Comparison Graph](comparison_graph.png)

## Quick Start

```python
from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained("CompactAI/Qwen2.5-0.5B-python-safe")
tokenizer = AutoTokenizer.from_pretrained("CompactAI/Qwen2.5-0.5B-python-safe")

inputs = tokenizer("Your prompt here", return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=100)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
```

## Technical Details

| Property | Value |
|----------|-------|
| Base Model | [Qwen/Qwen2.5-0.5B](https://huggingface.co/Qwen/Qwen2.5-0.5B) |
| Specialization | Python |
| Prune Mode | Safe |
| Weight Reduction | 1% weights pruned |

## License

This model inherits the license from the base model.