Andromeda-Code-0.5B / README.md
fiel1986's picture
Update README.md
e3cdea1 verified
|
Raw History Blame Contribute Delete
2.16 kB
---
license: apache-2.0
base_model: Qwen/Qwen2.5-0.5B-Instruct
language:
- es
pipeline_tag: text-generation
tags:
- codigo
- code
- destilacion
- response-distillation
- deepseek
- qwen
- espanol
library_name: transformers
---
![Andromeda](https://cdn-uploads.huggingface.co/production/uploads/69dd3cdd659f1094458104e3/J2THbjZIFL_TTNpOXu6Wg.jpeg)
# Andromeda-Code-0.5B 馃捇
Modelo de lenguaje de **0.5B par谩metros** especializado en **programaci贸n**,
destilado desde `deepseek-ai/deepseek-coder-1.3b-instruct` hacia
`Qwen/Qwen2.5-0.5B-Instruct`.
> DeepSeek y Qwen usan tokenizers distintos, as铆 que la destilaci贸n por logits
> no aplica directamente. Ac谩 se us贸 **response distillation**: el teacher genera
> respuestas de c贸digo, se filtran y limpian, y el student aprende con SFT sobre
> ese dataset sint茅tico.
## 馃幆 M茅todo de destilaci贸n
- **Teacher:** `deepseek-ai/deepseek-coder-1.3b-instruct`
- **Student:** `Qwen/Qwen2.5-0.5B-Instruct`
- **T茅cnica:** Response Distillation (Answer-Guided) + SFT
1. Generaci贸n de respuestas de c贸digo con el teacher
2. Limpieza: extracci贸n de bloques ```python, remoci贸n de texto introductorio
3. Conversi贸n al chat template de Qwen (formato JSONL)
4. Fine-tuning del student (3 epochs, lr=2e-5, bf16)
## 鈿狅笍 Estado experimental
Este modelo fue entrenado con un **dataset de prueba de solo 20 ejemplos**.
Sirve para validar el pipeline de destilaci贸n cross-tokenizer, pero su calidad
es limitada.
## 馃殌 Uso r谩pido
```python
from transformers import AutoModelForCausalLM, AutoTokenizer
model_id = "fiel1986/Andromeda-Code-0.5B"
tok = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
messages = [
{"role": "system", "content": "Sos un asistente experto en programacion."},
{"role": "user", "content": "Escribe una funcion en Python que verifique si un numero es primo"}
]
inputs = tok.apply_chat_template(messages, return_tensors="pt").to(model.device)
out = model.generate(inputs, max_new_tokens=256, do_sample=True, temperature=0.6)
print(tok.decode(out[0], skip_special_tokens=True))