diff --git a/modules/model_quant_nncf.py b/modules/model_quant_nncf.py index d748c081b..2b9129b07 100644 --- a/modules/model_quant_nncf.py +++ b/modules/model_quant_nncf.py @@ -35,15 +35,6 @@ class QuantizationMethod(str, Enum): NNCF = "nncf" -class ConvertToModule(torch.nn.Module): - def __init__(self, module): - super().__init__() - self.module = module - - def forward(self, *args, **kwargs): - return self.module(*args, **kwargs) - - class NNCFQuantizer(DiffusersQuantizer): r""" Diffusers Quantizer for NNCF @@ -187,7 +178,7 @@ class NNCFQuantizer(DiffusersQuantizer): class NNCFConfig(QuantizationConfigMixin): """ This is a wrapper class about all possible attributes and features that you can play with a model that has been - loaded using `quanto`. + loaded using `nncf`. Args: weights_dtype (`str`, *optional*, defaults to `"int8"`):