We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0a9381e commit 33d7f7dCopy full SHA for 33d7f7d
optimum/neuron/modeling_diffusion.py
@@ -992,7 +992,9 @@ def _export(
992
993
if cache_exist:
994
# load cache
995
- logger.info(f"Neuron cache found at {model_cache_dir}. If you want to recompile the model, please set `disable_neuron_cache=True`.")
+ logger.info(
996
+ f"Neuron cache found at {model_cache_dir}. If you want to recompile the model, please set `disable_neuron_cache=True`."
997
+ )
998
neuron_model = cls.from_pretrained(model_cache_dir, data_parallel_mode=data_parallel_mode)
999
# replace weights
1000
if not inline_weights_to_neff:
0 commit comments