bytestorm/SKIM-wikititles-kdd25

This is a LoRA-tuned checkpoint of Llama-2-7b for ranking tasks.

Model Details

• Base Model: Llama-2-7b
• Training Type: LoRA fine-tuning
• Task: Ranking/Retrieval
• Framework: PyTorch

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained("bytestorm/SKIM-wikititles-kdd25")
tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-hf")