yindo/flash-attention-prebuild-wheels-rocm
1
0
Fork 0
mirror of https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git synced 2026-07-01 01:37:53 -04:00
Code Issues Packages Projects Releases 2 Wiki Activity

update pytorch

Browse Source
This commit is contained in:
Junya Morioka
2025-08-10 16:56:59 +09:00
parent 8f23b8aae7
commit d7847260ec
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
get_torch_cuda_version.py
+1 -1
View File
@@ -9,7 +9,7 @@ support_cuda_versions = {
"2.5": [118, 121, 124],
"2.6": [118, 124, 126],
"2.7": [118, 126, 128],
"2.8": [128],
"2.8": [126, 128, 129],
}
cuda_version = int(sys.argv[1])