Mention the flash-attention restriction in the readme. (#1158)

huggingface · Oct 23, 2023 · 25c3cc4 · 25c3cc4
1 parent a11af79
commit 25c3cc4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/candle-examples/examples/stable-diffusion/README.md b/candle-examples/examples/stable-diffusion/README.md
@@ -50,6 +50,9 @@ cached.
 Enabling flash-attention requires both a feature flag, `--feature flash-attn`
 and using the command line flag `--use-flash-attn`.
 
+Note that flash-attention-v2 is only compatible with Ampere, Ada, or Hopper GPUs
+(e.g., A100/H100, RTX 3090/4090).
+
 ## Image to Image Pipeline
 ...