Update README.md

README.md

@@ -212,7 +212,7 @@ print(f"Transcription: {transcription}")
 ```
 
 __System requirements:__
-* GPUs: NVIDIA GeForce H100, L40S
+* GPUs: NVIDIA GeForce 4090, NVIDIA GeForce 5090, H100, L40S
 * CPU: AMD, Intel
 * Python: 3.8-3.12 (check dependencies for specific versions)
 
@@ -223,6 +223,15 @@ pip install thestage
 pip install 'thestage-elastic-models[nvidia]'
 pip install flash-attn==2.7.3 --no-build-isolation
 pip uninstall apex
+
+# or for blackwell support
+pip install 'thestage-elastic-models[blackwell]'
+pip install torch==2.7.0+cu128 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
+# please download the appropriate version of Wheels for your system from https://github.com/Zarrac/flashattention-blackwell-wheels-whl-ONLY-5090-5080-5070-5060-flash-attention-/releases/tag/FlashAttention
+mv flash_attn-2.7.4.post1-rtx5090-torch2.7.0cu128cxx11abiTRUE-cp311-linux_x86_64.whl flash_attn-2.7.4.post1-0rtx5090torch270cu128cxx11abiTRUE-cp311-cp311-linux_x86_64.whl
+pip install flash_attn-2.7.4.post1-0rtx5090torch270cu128cxx11abiTRUE-cp311-cp311-linux_x86_64.whl
+pip install tensorrt==10.11.0.33
+pip uninstall apex
 ```
 
 Then go to [app.thestage.ai](https://app.thestage.ai), login and generate API token from your profile page. Set up API token as follows:
@@ -260,6 +269,8 @@ Performance for transcribing audio (tps):
 |----------|---|----------|
 | H100 | 223.47 | 82.84 |
 | L40S | 210.67 | [TBD] |
+| GeForce RTX 4090 | 240 | 86.63 |
+| GeForce RTX 5090 | [TBD] | [TBD] |
 
 ## Links