Skip to content

Commit bd81f50

Browse files
committed
fix bnb
1 parent 645abc9 commit bd81f50

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/quantizers/bitsandbytes/bnb_quantizer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -493,7 +493,7 @@ def _process_model_after_weight_loading(self, model: "ModelMixin", **kwargs):
493493
model.is_8bit_serializable = self.is_serializable
494494
return model
495495

496-
# Copied from diffusers.quantizers.bitsandbytes.bnb_quantizer.BnB4BitDiffusersQuantizer._process_model_before_weight_loading
496+
# Copied from diffusers.quantizers.bitsandbytes.bnb_quantizer.BnB4BitDiffusersQuantizer._process_model_before_weight_loading with 4bit->8bit
497497
def _process_model_before_weight_loading(
498498
self,
499499
model: "ModelMixin",

0 commit comments

Comments
 (0)