Skip to content

Commit

Permalink
add back accidentally remove line
Browse files Browse the repository at this point in the history
  • Loading branch information
Titus-von-Koeller committed Feb 21, 2024
1 parent ad87fc4 commit b3a9bd5
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions bitsandbytes/nn/modules.py
Original file line number Diff line number Diff line change
Expand Up @@ -266,6 +266,7 @@ def _quantize(self, device):
quant_type=self.quant_type,
quant_storage=self.quant_storage,
)
self.data = w_4bit
self.quant_state = quant_state
if self.module is not None:
self.module.quant_state = quant_state
Expand Down

0 comments on commit b3a9bd5

Please sign in to comment.