from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline

tokenizer = "mistralai/Mistral-7B-Instruct-v0.2"
model_id = "Sefika/mistral_fewrel_10_8"
tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=True)
model = AutoModelForCausalLM.from_pretrained(
    model_id,
    device_map="auto",
    load_in_4bit=True,  # Requires bitsandbytes
    torch_dtype="auto"
)

Testing Data

Repository: [More Information Needed]
Paper [optional]: [More Information Needed]
Demo [optional]: [More Information Needed]

FewRel

BibTeX: The paper "Large Language Models for Continual Relation Extraction" is submitted to Springer Machine Learning journal

Model Card Contact

sefika efeoglu