nvidia/Minitron-8B-Base
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Update installation instructions

Browse Source
This commit is contained in:
Saurav Muralidharan
2024-08-20 11:03:27 -07:00
parent c4b8f786c9
commit 70fa5997af
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
README.md
View File

@@ -48,11 +48,10 @@ It also uses Grouped-Query Attention (GQA) and Rotary Position Embeddings (RoPE)
## Usage ## Usage
The [pull request](https://github.com/huggingface/transformers/pull/32495) to support this model in Hugging Face Transformers is under review and is expected to be merged soon. In the meantime, please follow the installation instructions below: Support for this model will be added in the upcoming `transformers` release. In the meantime, please install the library from source:
``` ```
$ git clone -b aot/head_dim_rope --single-branch https://github.com/suiyoubi/transformers.git && cd transformers pip install git+https://github.com/huggingface/transformers
$ pip install -e .
``` ```
The following code provides an example of how to load the Minitron-8B model and use it to perform text generation. The following code provides an example of how to load the Minitron-8B model and use it to perform text generation.