fix

echarlaix · echarlaix · commit 416151755421 · 2024-01-30T16:20:38.000+01:00
diff --git a/optimum/intel/neural_compressor/modeling_base.py b/optimum/intel/neural_compressor/modeling_base.py
@@ -144,7 +144,7 @@ def _from_pretrained(
 
         if isinstance(state_dict, torch.jit.ScriptModule):
             raise RuntimeError(
-                f"`{cls.__name__}` does not support the loading TorchScript model, to load your model please use {cls.__name__.replace('INC', 'IPEX')}` instead."
+                f"`{cls.__name__}` does not support the loading of TorchScript models, to load your model please use `{cls.__name__.replace('INC', 'IPEX')}` instead."
             )
 
         q_config = state_dict.get("best_configure", None)

Original file line number	Diff line number	Diff line change
`@@ -144,7 +144,7 @@ def _from_pretrained(`
`144`	`144`
`145`	`145`	`if isinstance(state_dict, torch.jit.ScriptModule):`
`146`	`146`	`raise RuntimeError(`
`147`		- f"`{cls.__name__}` does not support the loading TorchScript model, to load your model please use {cls.__name__.replace('INC', 'IPEX')}` instead."
	`147`	+ f"`{cls.__name__}` does not support the loading of TorchScript models, to load your model please use `{cls.__name__.replace('INC', 'IPEX')}` instead."
`148`	`148`	`)`
`149`	`149`
`150`	`150`	`q_config = state_dict.get("best_configure", None)`