Skip to content

Commit 9471404

Browse files
authored
Merge pull request #1683 from NeuralFault/universal-rocm
Update win-rocm bnb wheel
2 parents faca7f4 + 2db45c4 commit 9471404

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

StabilityMatrix.Core/Models/PackageModification/InstallWindowsRocmPackageCommandStep.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ IRocmPackageHelper rocmPackageHelper
2525
) : IPackageStep
2626
{
2727
private const string BitsAndBytesWheelUrl =
28-
"https://github.com/0xDELUXA/bitsandbytes_win_rocm/releases/download/0.50.0.dev0-py3-rocm7-win_amd64_all/bitsandbytes-0.50.0.dev0-cp312-cp312-win_amd64.whl";
28+
"https://github.com/0xDELUXA/bitsandbytes_win_rocm/releases/download/0.50.0.dev0-py3.12-rocm7.15-win_amd64_all/bitsandbytes-0.50.0.dev0-cp312-cp312-win_amd64.whl";
2929
private const string AmdAiterWheelUrl =
3030
"https://github.com/0xDELUXA/flash-attention/releases/download/v2.8.4_win-rocm/amd_aiter-0.0.0-py3-none-win_amd64.whl";
3131
private const string FlashAttentionWheelUrl =

StabilityMatrix.Core/Models/Rocm/InvokeAiWindowsRocmProfile.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ public class InvokeAiWindowsRocmProfile : RocmPackageProfile
1515

1616
// Replace upstream bitsandbytes with ROCm-aware bitsandbytes for TheRock ROCm 7.13 on Windows
1717
private const string BitsAndBytesWheelUrl =
18-
"https://github.com/0xDELUXA/bitsandbytes_win_rocm/releases/download/0.50.0.dev0-py3-rocm7-win_amd64_all/bitsandbytes-0.50.0.dev0-cp312-cp312-win_amd64.whl";
18+
"https://github.com/0xDELUXA/bitsandbytes_win_rocm/releases/download/0.50.0.dev0-py3.12-rocm7.15-win_amd64_all/bitsandbytes-0.50.0.dev0-cp312-cp312-win_amd64.whl";
1919

2020
public static RocmPackageProfile CreateInstallProfile(PyVersion pyVersion)
2121
{

0 commit comments

Comments
 (0)