[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):
|
) or self.is_layer_skipped(prefix, self.packed_modules_mapping):
|
||||||
return UnquantizedLinearMethod()
|
return UnquantizedLinearMethod()
|
||||||
layer.scheme = self.get_linear_scheme(layer, prefix_in_quant_config)
|
layer.scheme = self.get_linear_scheme(layer, prefix_in_quant_config)
|
||||||
|
if layer.scheme is None:
|
||||||
|
return UnquantizedLinearMethod()
|
||||||
return ModelSlimLinearMethod(self)
|
return ModelSlimLinearMethod(self)
|
||||||
elif isinstance(layer, FusedMoE):
|
elif isinstance(layer, FusedMoE):
|
||||||
layer.scheme = self.get_moe_scheme(layer, prefix)
|
layer.scheme = self.get_moe_scheme(layer, prefix)
|
||||||
|
|||||||
Reference in New Issue
Block a user