CIR-AMS commited on
Commit
d8dd1e3
1 Parent(s): 1769b31

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -31,7 +31,7 @@ pip install numpy pandas
31
  # python scripts/run_rm.py --model={modelpath}
32
  ```
33
 
34
- See btrm-qwen2.py for the code to use the model
35
  ```bash
36
  from transformers import AutoTokenizer, pipeline
37
  import torch
 
31
  # python scripts/run_rm.py --model={modelpath}
32
  ```
33
 
34
+ For the code that uses the model, see the following, which is how the model is commonly loaded
35
  ```bash
36
  from transformers import AutoTokenizer, pipeline
37
  import torch