Skip to content

Commit 33d7f7d

Browse files
JingyaHuangdacorvo
authored andcommitted
fix: style
1 parent 0a9381e commit 33d7f7d

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

optimum/neuron/modeling_diffusion.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -992,7 +992,9 @@ def _export(
992992

993993
if cache_exist:
994994
# 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`.")
995+
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+
)
996998
neuron_model = cls.from_pretrained(model_cache_dir, data_parallel_mode=data_parallel_mode)
997999
# replace weights
9981000
if not inline_weights_to_neff:

0 commit comments

Comments
 (0)