Qwen-1.8B Ecuador QA
Modelo fine-tuneado de Qwen-1.8B para responder preguntas en español sobre Ecuador.
Descripción
- Modelo Base: Qwen/Qwen-1_8B
- Dataset: Jnaranjo/EcuadorWeb-QA (10,597 ejemplos)
- Idioma: Español (contexto ecuatoriano)
Uso
Con Transformers
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained("Jnaranjo/qwen-1.8b-ecuador-qa", trust_remote_code=True)
tokenizer = AutoTokenizer.from_pretrained("Jnaranjo/qwen-1.8b-ecuador-qa", trust_remote_code=True)
if tokenizer.pad_token is None:
tokenizer.pad_token = tokenizer.eos_token
pregunta = "¿Dónde se encuentra el estudio 108 Yoga?"
prompt = f"Question: {pregunta}\nAnswer:"
inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
outputs = model.generate(**inputs, max_new_tokens=256, temperature=0.7, do_sample=True)
respuesta = tokenizer.decode(outputs[0], skip_special_tokens=True)[len(prompt):].strip()
print(f"Respuesta: {respuesta}")
- Downloads last month
- 12