CoolFace
Modelpublic

PraneelChetty-1/smollm2-135m-query-expansion

sourceHugging Faceapache-2.0updated 5mo agoView on Hugging Face
2likes76downloads
Model Card

SmolLM2-135M — Query Expansion (Fine-tuned)

<p align="center"> <img src="./query-expansion.png" alt="Query Expansion Architecture" width="600"/> </p>

Fine-tuned version of SmolLM2-135M-Instruct for the task of search query expansion.

Training

  • —Base model: HuggingFaceTB/SmolLM2-135M-Instruct
  • —Method: LoRA (merged into base weights at export)
  • —Task: Query expansion / reformulation

Quick usage

python
from transformers import AutoTokenizer, pipeline

tokenizer = AutoTokenizer.from_pretrained("PraneelChetty-1/smollm2-135m-query-expansion")
pipe = pipeline("text-generation", model="PraneelChetty-1/smollm2-135m-query-expansion", device_map="auto")

messages = [{"role": "user", "content": "yield monitoring systems"}]
prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)

out = pipe(prompt, max_new_tokens=80, do_sample=True, temperature=0.7)
print(out[0]["generated_text"][len(prompt):])