@ -22,17 +22,19 @@ import os
USE_STABLE_EMBEDDING = False
try :
import bitsandbytes as bnb
OVERRIDE_LINEAR = False
OVERRIDE_EMBEDDING = Tru e
OVERRIDE_ADAM = Tru e
OVERRIDE_ADAMW = Tru e
OVERRIDE_EMBEDDING = Fals e
OVERRIDE_ADAM = Fals e
OVERRIDE_ADAMW = Fals e
USE_STABLE_EMBEDDING = os . environ . get ( ' BITSANDBYTES_USE_STABLE_EMBEDDING ' , ' 1 ' if USE_STABLE_EMBEDDING else ' 0 ' ) == ' 1 '
OVERRIDE_LINEAR = os . environ . get ( ' BITSANDBYTES_OVERRIDE_LINEAR ' , ' 1 ' if OVERRIDE_LINEAR else ' 0 ' ) == ' 1 '
OVERRIDE_EMBEDDING = os . environ . get ( ' BITSANDBYTES_OVERRIDE_EMBEDDING ' , ' 1 ' if OVERRIDE_EMBEDDING else ' 0 ' ) == ' 1 '
OVERRIDE_ADAM = os . environ . get ( ' BITSANDBYTES_OVERRIDE_ADAM ' , ' 1 ' if OVERRIDE_ADAM else ' 0 ' ) == ' 1 '
OVERRIDE_ADAMW = os . environ . get ( ' BITSANDBYTES_OVERRIDE_ADAMW ' , ' 1 ' if OVERRIDE_ADAMW else ' 0 ' ) == ' 1 '
if OVERRIDE_LINEAR or OVERRIDE_EMBEDDING or OVERRIDE_ADAM or OVERRIDE_ADAMW :
import bitsandbytes as bnb
except Exception as e :
OVERRIDE_LINEAR = False
OVERRIDE_EMBEDDING = False