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)

Query Expansion Architecture

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):])
Description
Model synced from source: PraneelChetty-1/smollm2-135m-query-expansion
Readme 765 KiB