pkoloveas commited on
Commit
a2f922c
·
verified ·
1 Parent(s): 03ec37a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -9
README.md CHANGED
@@ -17,7 +17,7 @@ tags:
17
  ---
18
 
19
 
20
- # Qwen2.5-14-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,16 +33,16 @@ A fine-tuned model for Citation Intent Classification, based on [Qwen 2.5 14B In
33
  ## Quickstart
34
 
35
  ```python
36
- # from transformers import AutoModelForCausalLM, AutoTokenizer
37
 
38
- # model_name = "Qwen/Qwen2.5-14B-Instruct"
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 #
 
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 #