TinyLlama-1.1B-Chat-v1.0-python-safe

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

This model is a conservatively pruned version of TinyLlama/TinyLlama-1.1B-Chat-v1.0.

Performance Comparison

Category	Original	Pruned	Change
Python	0.0%	0.0% ⭐	→
Html	0.0%	0.0%	→
Trivia	35.7%	35.7%	→
Math	0.0%	0.0%	→
Reasoning	0.0%	0.0%	→
Medical	50.0%	50.0%	→
Linux	80.0%	80.0%	→
Writing	16.7%	16.7%	→

Average: 22.8% → 22.8% (+0.0%)

Quick Start

from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained("CompactAI/TinyLlama-1.1B-Chat-v1.0-python-safe")
tokenizer = AutoTokenizer.from_pretrained("CompactAI/TinyLlama-1.1B-Chat-v1.0-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	TinyLlama/TinyLlama-1.1B-Chat-v1.0
Specialization	Python
Prune Mode	Safe
Weight Reduction	1% weights pruned

License

This model inherits the license from the base model.