lilblueyes commited on
Commit
2eb805a
·
1 Parent(s): 0fd1a4f

Add optional GPU flash attention dependencies

Browse files
Files changed (2) hide show
  1. README.md +10 -0
  2. requirements-gpu.txt +4 -0
README.md CHANGED
@@ -12,3 +12,13 @@ pinned: false
12
  # ASL TTS Test
13
 
14
  MVP Gradio pour tester la brique TTS, llama.cpp et une interface custom avant de brancher le pipeline vidéo ASL.
 
 
 
 
 
 
 
 
 
 
 
12
  # ASL TTS Test
13
 
14
  MVP Gradio pour tester la brique TTS, llama.cpp et une interface custom avant de brancher le pipeline vidéo ASL.
15
+
16
+ ## GPU dependencies
17
+
18
+ `flash-attn` is only useful on a CUDA GPU Space with compatible PyTorch/CUDA versions.
19
+ Keep the default `requirements.txt` for CPU Spaces. If the Space is moved to a compatible
20
+ GPU runtime, install the GPU dependency set instead:
21
+
22
+ ```bash
23
+ pip install -r requirements-gpu.txt --no-build-isolation
24
+ ```
requirements-gpu.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ -r requirements.txt
2
+
3
+ ninja
4
+ flash-attn