CompactAI's picture
Upload folder using huggingface_hub
c75b72d verified
---
license: apache-2.0
tags:
- pruned
- python
- optimized
- wanda
- activation-pruning
base_model: Qwen/Qwen3-0.6B
pipeline_tag: text-generation
---
# Qwen3-0.6B-python-extra-heavy
> 🎯 **PYTHON-optimized** | πŸ“¦ **Extra Heavy** pruning | ⚑ **20% weights pruned**
This model is a **extremely pruned** version of [Qwen/Qwen3-0.6B](https://huggingface.co/Qwen/Qwen3-0.6B), specialized for **PYTHON** tasks using activation-aware weight pruning (Wanda-style).
## ✨ Key Features
- **Specialization**: Optimized for Python tasks
- **Pruning Method**: Wanda-style (|W| Γ— |activation|) importance scoring
- **Size Reduction**: 20% weights pruned
- **Use Case**: Maximum size reduction, best for edge deployment
## πŸ“Š Performance Comparison
| Category | Original | Pruned | Change |
|----------|----------|--------|--------|
| **Python** | 40.0% | 20.0% ⭐ | ↓ 20.0% |
| Html | 0.0% | 0.0% | β†’ |
| Trivia | 80.0% | 66.7% | ↓ 13.3% |
| Math | 100.0% | 100.0% | β†’ |
| Reasoning | N/A | N/A | |
| Medical | 93.3% | 93.3% | β†’ |
| Linux | 100.0% | 93.3% | ↓ 6.7% |
| Writing | 33.3% | 13.3% | ↓ 20.0% |
**Average**: 63.8% β†’ 55.2% (-8.6%)
**Python Retention**: 50.0% of original performance
![Comparison Graph](comparison_graph.png)
## πŸš€ Quick Start
```python
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained("CompactAI/Qwen3-0.6B-python-extra-heavy")
tokenizer = AutoTokenizer.from_pretrained("CompactAI/Qwen3-0.6B-python-extra-heavy")
# Example usage
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/Qwen3-0.6B](https://huggingface.co/Qwen/Qwen3-0.6B) |
| Specialization | Python |
| Prune Mode | Extra Heavy |
| Pruning Method | Activation-based weight pruning (Wanda) |
| Weight Reduction | 20% weights pruned |
## πŸ”— Related Models
This model is part of the **Qwen3-0.6B** pruned model collection. Other variants:
- Extra-light (minimal pruning)
- Light
- Medium-light
- Medium
- Medium-heavy
- Heavy
- Extra-heavy (maximum compression)
## πŸ“œ License
This model inherits the license from the base model [Qwen/Qwen3-0.6B](https://huggingface.co/Qwen/Qwen3-0.6B).
---
*Generated by ZANNPS [Zeto Automatic Neural Network Pruning System]*