Update device_type.py
This commit is contained in:
parent
2697047c20
commit
cc9809d2ea
1 changed files with 1 additions and 1 deletions
|
|
@ -81,7 +81,7 @@ if DEVICE_TYPE == "hip":
|
|||
if "blocksize = 64 if not HIP_ENVIRONMENT else 128" in inspect.getsource(Params4bit):
|
||||
ALLOW_PREQUANTIZED_MODELS = False
|
||||
import bitsandbytes
|
||||
ALLOW_BITSANDBYTES = Version(bitsandbytes.__version__) >= Version("0.48.2.dev0")
|
||||
ALLOW_BITSANDBYTES = Version(bitsandbytes.__version__) > Version("0.48.2.dev0")
|
||||
except:
|
||||
pass
|
||||
pass
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue