nvidia
/

Hymba-1.5B-Base

Text Generation

Model card Files Files and versions Community

YongganFu commited on Nov 26, 2024

Commit

ac59e23

·

verified ·

1 Parent(s): 936bb7f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -63,7 +63,7 @@ Features of this architecture:
 ### Step 1: Environment Setup
-Since Hymba-1.5B-Instruct employs [FlexAttention](https://pytorch.org/blog/flexattention/), which relies on Pytorch2.5 and other related dependencies, we provide two ways to setup the environment:
 - **[Local install]** Install the related packages using our provided `setup.sh` (support CUDA 12.1/12.4):

 ### Step 1: Environment Setup
+Since Hymba-1.5B-Base employs [FlexAttention](https://pytorch.org/blog/flexattention/), which relies on Pytorch2.5 and other related dependencies, we provide two ways to setup the environment:
 - **[Local install]** Install the related packages using our provided `setup.sh` (support CUDA 12.1/12.4):