yingbei commited on
Commit
4ddae95
·
verified ·
1 Parent(s): 4d84409

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -72,7 +72,7 @@ The model was post-trained (freeze tuned & DPO) on a proprietary dataset consist
72
  You can use the model with the Hugging Face `transformers` and the rubra library [rubra-tools](https://github.com/rubra-ai/rubra-tools) as follows:
73
 
74
  ```
75
- pip install rubra_tools torch==2.3.0 transformers
76
  ```
77
 
78
  You also need Node.js and npm installed. Once you do, install the `jsonrepair` package - it's used to fix some rare hallucinations by the model.
 
72
  You can use the model with the Hugging Face `transformers` and the rubra library [rubra-tools](https://github.com/rubra-ai/rubra-tools) as follows:
73
 
74
  ```
75
+ pip install rubra_tools torch==2.3.0 transformers accelerate
76
  ```
77
 
78
  You also need Node.js and npm installed. Once you do, install the `jsonrepair` package - it's used to fix some rare hallucinations by the model.