Update README.md

README.md

@@ -4,44 +4,8 @@ license: bsd-3-clause
 This repo provides the wheel for Flash Attention 3 for Hopper architecture for the current default cuda version on ZeroGPU spaces,
 which at the moment is Torch version: 2.8.0+cu128. 
 
-To download and install the code in your Space you can copy/paste the code snippet below
+You can add this to the requirements.txt file of your space
 
-```Python
-import subprocess, site, importlib
-from huggingface_hub import hf_hub_download
-
-# Re-discover all .pth/.egg-link files
-for sitedir in site.getsitepackages():
-    site.addsitedir(sitedir)
-
-# Clear caches so importlib will pick up new modules
-importlib.invalidate_caches()
-
-def sh(cmd): subprocess.check_call(cmd, shell=True)
-
-flash_attention_installed = False
-
-try:
-    print("Attempting to download and install FlashAttention wheel...")
-    flash_attention_wheel = hf_hub_download(
-            repo_id="alexnasa/flash-attn-3",
-            repo_type="model",
-            filename="128/flash_attn_3-3.0.0b1-cp39-abi3-linux_x86_64.whl",
-        )
-
-    sh(f"pip install {flash_attention_wheel}")
-
-    # tell Python to re-scan site-packages now that the egg-link exists
-    import importlib, site; site.addsitedir(site.getsitepackages()[0]); importlib.invalidate_caches()
-
-    flash_attention_installed = True
-    print("FlashAttention installed successfully.")
-
-except Exception as e:
-    print(f"⚠️ Could not install FlashAttention: {e}")
-    print("Continuing without FlashAttention...")
-
-import torch
-print(f"Torch version: {torch.__version__}")
-print(f"FlashAttention available: {flash_attention_installed}")
-```
+```
+flash-attn-3 @ https://huggingface.co/alexnasa/flash-attn-3/resolve/main/128/flash_attn_3-3.0.0b1-cp39-abi3-linux_x86_64.whl
+```