Logits-Based Finetuning
Paper • 2505.24461 • Published • 1
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("JingyaoLi/ScienceLLaMA-3b")
model = AutoModelForCausalLM.from_pretrained("JingyaoLi/ScienceLLaMA-3b")
messages = [
{"role": "user", "content": "Who are you?"},
]
inputs = tokenizer.apply_chat_template(
messages,
add_generation_prompt=True,
tokenize=True,
return_dict=True,
return_tensors="pt",
).to(model.device)
outputs = model.generate(**inputs, max_new_tokens=40)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))• 🤗 Data • 🤗 ScienceLLaMA-3B • 🤗 ScienceLLaMA-1B • 🐱 Code • 📃 Paper
This model is a fine-tuned with Logits-Based Finetuning on the JingyaoLi/Science-Logits-1.2M, which integrates the strengths of supervised learning and knowledge distillation by combining teacher logits with ground truth labels. This preserves both correctness and linguistic diversity.
The following hyperparameters were used during training:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="JingyaoLi/ScienceLLaMA-3b") messages = [ {"role": "user", "content": "Who are you?"}, ] pipe(messages)