From 9fec8a83240609d1505eb6c676bb9a9c2fbb6f96 Mon Sep 17 00:00:00 2001 From: Gavi Narra Date: Mon, 3 Jun 2024 09:26:26 -0400 Subject: [PATCH] Update README.md --- kosmos-2.5/README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/kosmos-2.5/README.md b/kosmos-2.5/README.md index 02bb64077..244332bdd 100644 --- a/kosmos-2.5/README.md +++ b/kosmos-2.5/README.md @@ -37,6 +37,7 @@ wget https://huggingface.co/microsoft/kosmos-2.5/resolve/main/ckpt.pt?download=t The code uses [Flash Attention2](https://github.com/Dao-AILab/flash-attention), so it only runs on Ampere, Ada, or Hopper GPUs (e.g., A100, RTX 3090, RTX 4090, H100). ``` bash git clone https://github.com/microsoft/unilm.git +cd unilm cd kosmos-2.5 pip install -r requirements.txt ```