Skip to content

Commit 536ebfb

Browse files
MaxusmustiMaxusmusti
authored
Add kernels>0.9.0 to CUDA requirements (#658)
Signed-off-by: Mustafa Eyceoz <[email protected]>
1 parent 8f0c6ed commit 536ebfb

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

‎requirements-cuda.txt‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
flash-attn>=2.4.0
22
bitsandbytes>=0.43.1
3+
kernels>=0.9.0
34

45
# required for FSDP updates
56
accelerate>=0.34.2

0 commit comments

Comments
 (0)