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 |
{
|