Hugging Face's logo

Sefika
/
mistral_fewrel_10_10

Text Generation
Transformers
Safetensors
mistral
trl
sft
text-generation-inference
4-bit precision
bitsandbytes
Model card Files Community
mistral_fewrel_10_10 / README.md
Sefika's picture
Sefika
Update README.md
3601a80 verified
preview code
|
raw
history blame contribute delete
1.38 kB
metadata
library_name: transformers
tags:
  - trl
  - sft

Model Card for Model ID

Model Details

Model Description

This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.

  • Developed by: Sefika

  • Language(s) (NLP): EN

  • License: MIT

  • Finetuned from model [optional]: mistralai/Mistral-7B-Instruct-v0.2

Model Sources [optional]

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

Direct Use 

from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline

tokenizer = "mistralai/Mistral-7B-Instruct-v0.2"
model_id = "Sefika/mistral_fewrel_10_10"
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

FewRel

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

Model Card Contact

sefika efeoglu