minhtt32 commited on
Commit
c720d49
1 Parent(s): b3c885c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -7,14 +7,14 @@ language:
7
  ---
8
  <div align="center">
9
 
10
- # UniNER-7B-all
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 our [paper](https://arxiv.org/abs/2308.03279) for more information.
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="/media/4TB_1/minhtt/Documents/test/axolotl/lora-out/merged",
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
  {