diff --git a/bitsandbytes/cextension.py b/bitsandbytes/cextension.py index a096d0d51..cfeaf4f44 100644 --- a/bitsandbytes/cextension.py +++ b/bitsandbytes/cextension.py @@ -119,7 +119,7 @@ def get_native_library() -> BNBNativeLibrary: if torch.version.hip: hip_major, hip_minor = map(int, torch.version.hip.split(".")[0:2]) HIP_ENVIRONMENT, BNB_HIP_VERSION = True, hip_major * 100 + hip_minor - BNB_HIP_VERSION_SHORT = str(hip_major) + str(hip_minor) + BNB_HIP_VERSION_SHORT = f"{hip_major}{hip_minor}" else: HIP_ENVIRONMENT, BNB_HIP_VERSION = False, 0 BNB_HIP_VERSION_SHORT = ""