mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 17:24:32 +02:00
SDNQ add explicit model.quantization_method = QuantizationMethod.SDNQ
This commit is contained in:
@@ -790,6 +790,7 @@ class SDNQQuantizer(DiffusersQuantizer, HfQuantizer):
|
||||
except Exception:
|
||||
pass
|
||||
model.quantization_config = self.quantization_config
|
||||
model.quantization_method = QuantizationMethod.SDNQ
|
||||
|
||||
def _process_model_after_weight_loading(self, model, **kwargs): # pylint: disable=unused-argument
|
||||
if self.quantization_config.is_training:
|
||||
|
||||
Reference in New Issue
Block a user