[NPU][Bugfix] Fix a ModelSlim loading failure (#29029)
This commit is contained in:
@@ -177,6 +177,8 @@ class ModelSlimConfig(QuantizationConfig):
|
||||
) or self.is_layer_skipped(prefix, self.packed_modules_mapping):
|
||||
return UnquantizedLinearMethod()
|
||||
layer.scheme = self.get_linear_scheme(layer, prefix_in_quant_config)
|
||||
if layer.scheme is None:
|
||||
return UnquantizedLinearMethod()
|
||||
return ModelSlimLinearMethod(self)
|
||||
elif isinstance(layer, FusedMoE):
|
||||
layer.scheme = self.get_moe_scheme(layer, prefix)
|
||||
|
||||
Reference in New Issue
Block a user