Model: PraneelChetty-1/smollm2-135m-query-expansion Source: Original Platform
base_model, library_name, license, tags, language
| base_model | library_name | license | tags | language | ||||||
|---|---|---|---|---|---|---|---|---|---|---|
| HuggingFaceTB/SmolLM2-135M-Instruct | transformers | apache-2.0 |
|
|
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):])
Description
