Update README.md
Browse files
README.md
CHANGED
@@ -17,7 +17,7 @@ tags:
|
|
17 |
---
|
18 |
|
19 |
|
20 |
-
# Qwen2.5-
|
21 |
|
22 |
A fine-tuned model for Citation Intent Classification, based on [Qwen 2.5 14B Instruct](https://huggingface.co/Qwen/Qwen2.5-14B-Instruct) and trained on the [SciCite](https://huggingface.co/datasets/allenai/scicite) dataset.
|
23 |
|
@@ -33,16 +33,16 @@ A fine-tuned model for Citation Intent Classification, based on [Qwen 2.5 14B In
|
|
33 |
## Quickstart
|
34 |
|
35 |
```python
|
36 |
-
|
37 |
|
38 |
-
|
39 |
|
40 |
-
|
41 |
-
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
-
|
46 |
|
47 |
system_prompt = """
|
48 |
# CONTEXT #
|
|
|
17 |
---
|
18 |
|
19 |
|
20 |
+
# Qwen2.5-14B-CIC-SciCite
|
21 |
|
22 |
A fine-tuned model for Citation Intent Classification, based on [Qwen 2.5 14B Instruct](https://huggingface.co/Qwen/Qwen2.5-14B-Instruct) and trained on the [SciCite](https://huggingface.co/datasets/allenai/scicite) dataset.
|
23 |
|
|
|
33 |
## Quickstart
|
34 |
|
35 |
```python
|
36 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
37 |
|
38 |
+
model_name = "sknow-lab/Qwen2.5-14B-CIC-SciCite"
|
39 |
|
40 |
+
model = AutoModelForCausalLM.from_pretrained(
|
41 |
+
model_name,
|
42 |
+
torch_dtype="auto",
|
43 |
+
device_map="auto"
|
44 |
+
)
|
45 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
46 |
|
47 |
system_prompt = """
|
48 |
# CONTEXT #
|