File size: 1,392 Bytes
5e35646 3c970f6 5e35646 3c970f6 5e35646 3c970f6 5e35646 3c970f6 5e35646 3c970f6 5e35646 3c970f6 5e35646 3c970f6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
---
library_name: transformers
tags:
- trl
- sft
---
# Model Card for Model ID
<!-- Provide a quick summary of what the model is/does. -->
## Model Details
### Model Description
<!-- Provide a longer summary of what this model is. -->
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]:** https://huggingface.co/meta-llama/Llama-2-7b-chat-hf
### Model Sources [optional]
<!-- Provide the basic links for the model. -->
- **Repository:** [More Information Needed]
- **Paper [optional]:** [More Information Needed]
- **Demo [optional]:** [More Information Needed]
### Direct Use
```python
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
tokenizer = "meta-llama/Llama-2-7b-chat-hf"
model_id = "CRE_llama_fewrel_1_memory_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
FewRel
**BibTeX:**
The paper "Large Language Models for Continual Relation Extraction" is submitted to Springer Machine Learning journal
## Model Card Contact
sefika efeoglu
|