johnrachwanpruna commited on
Commit
5ab374e
1 Parent(s): 302d825

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -50,7 +50,7 @@ These are several general ways to use the DBRX models:
50
  Getting started with DBRX models is easy with the `transformers` library. The model requires ~264GB of RAM and the following packages:
51
 
52
  ```bash
53
- pip install "transformers>=4.39.2" "tiktoken>=0.6.0"
54
  ```
55
 
56
  If you'd like to speed up download time, you can use the `hf_transfer` package as described by Huggingface [here](https://huggingface.co/docs/huggingface_hub/en/guides/download#faster-downloads).
 
50
  Getting started with DBRX models is easy with the `transformers` library. The model requires ~264GB of RAM and the following packages:
51
 
52
  ```bash
53
+ pip install "torch==2.4.0" "transformers>=4.39.2" "tiktoken>=0.6.0" "bitsandbytes"
54
  ```
55
 
56
  If you'd like to speed up download time, you can use the `hf_transfer` package as described by Huggingface [here](https://huggingface.co/docs/huggingface_hub/en/guides/download#faster-downloads).