File size: 2,420 Bytes
384a0e6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
---
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-heavy

> 🎯 **PYTHON-optimized** | πŸ“¦ **Heavy** pruning | ⚑ **15% weights pruned**

This model is a **aggressively 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**: 15% weights pruned
- **Use Case**: Maximum compression while maintaining usability

## πŸ“Š Performance Comparison

| Category | Original | Pruned | Change |
|----------|----------|--------|--------|
| **Python** | 40.0% | 40.0% ⭐ | β†’ |
| Html | 0.0% | 0.0% | β†’ |
| Trivia | 80.0% | 73.3% | ↓ 6.7% |
| Math | 100.0% | 100.0% | β†’ |
| Reasoning | N/A | N/A |  |
| Medical | 93.3% | 93.3% | β†’ |
| Linux | 100.0% | 100.0% | β†’ |
| Writing | 33.3% | 33.3% | β†’ |

**Average**: 63.8% β†’ 62.9% (-1.0%)

**Python Retention**: 100.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-heavy")
tokenizer = AutoTokenizer.from_pretrained("CompactAI/Qwen3-0.6B-python-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 | Heavy |
| Pruning Method | Activation-based weight pruning (Wanda) |
| Weight Reduction | 15% 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]*