diff --git a/codes/torch_intermediary/__init__.py b/codes/torch_intermediary/__init__.py index 5e050226..70556a2c 100644 --- a/codes/torch_intermediary/__init__.py +++ b/codes/torch_intermediary/__init__.py @@ -20,7 +20,7 @@ OVERRIDE_ADAMW = False # True USE_STABLE_EMBEDDING = True try: - import bitsandbytes as bnb + import bitsandbytes as bnb OVERRIDE_LINEAR = False OVERRIDE_EMBEDDING = False OVERRIDE_ADAM = True