Qwen-1.8B Ecuador QA

Modelo fine-tuneado de Qwen-1.8B para responder preguntas en español sobre Ecuador.

Descripción

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
Safetensors
Model size
1.84B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train Jnaranjo/qwen-1.8b-ecuador-qa

Evaluation results