smollm2-135m-query-expansion/README.md

base_model, library_name, license, tags, language

base_model	library_name	license	tags	language
HuggingFaceTB/SmolLM2-135M-Instruct	transformers	apache-2.0	text-generation query-expansion fine-tuned smollm2 lora	en

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

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

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):])