diff --git a/bitsandbytes/nn/modules.py b/bitsandbytes/nn/modules.py index cdeaebc27..f28ef651f 100755 --- a/bitsandbytes/nn/modules.py +++ b/bitsandbytes/nn/modules.py @@ -447,7 +447,7 @@ def __init__( ) # self.persistent_buffers = [] # TODO consider as way to save quant state self.compute_dtype = compute_dtype - self.compute_type_is_set = False + self.compute_type_is_set = False if compute_dtype is None else True self.ipex_linear_is_set = False self.quant_state = None self.quant_storage = quant_storage