Update README.md
Browse files
README.md
CHANGED
|
@@ -7,14 +7,14 @@ language:
|
|
| 7 |
---
|
| 8 |
<div align="center">
|
| 9 |
|
| 10 |
-
#
|
| 11 |
</div>
|
| 12 |
|
| 13 |
This model is finetuned from [TinyLLama](https://huggingface.co/TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T).
|
| 14 |
|
| 15 |
It is trained on ChatGPT-generated [Pile-NER-type data](https://huggingface.co/datasets/Universal-NER/Pile-NER-type).
|
| 16 |
|
| 17 |
-
Check
|
| 18 |
|
| 19 |
|
| 20 |
### How to use
|
|
@@ -28,7 +28,7 @@ Do check the [TinyLlama](https://github.com/jzhang38/TinyLlama) github page for
|
|
| 28 |
import torch
|
| 29 |
from transformers import pipeline
|
| 30 |
|
| 31 |
-
pipe = pipeline("text-generation", model="/
|
| 32 |
torch_dtype=torch.bfloat16, device_map="auto")
|
| 33 |
messages = [
|
| 34 |
{
|
|
|
|
| 7 |
---
|
| 8 |
<div align="center">
|
| 9 |
|
| 10 |
+
# tiny-universal-NER
|
| 11 |
</div>
|
| 12 |
|
| 13 |
This model is finetuned from [TinyLLama](https://huggingface.co/TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T).
|
| 14 |
|
| 15 |
It is trained on ChatGPT-generated [Pile-NER-type data](https://huggingface.co/datasets/Universal-NER/Pile-NER-type).
|
| 16 |
|
| 17 |
+
Check this [paper](https://arxiv.org/abs/2308.03279) for more information.
|
| 18 |
|
| 19 |
|
| 20 |
### How to use
|
|
|
|
| 28 |
import torch
|
| 29 |
from transformers import pipeline
|
| 30 |
|
| 31 |
+
pipe = pipeline("text-generation", model="LR-AI-Labs/tiny-universal-NER",
|
| 32 |
torch_dtype=torch.bfloat16, device_map="auto")
|
| 33 |
messages = [
|
| 34 |
{
|