From 96191cee565fb4c5b6689f9abffa8649c07956b5 Mon Sep 17 00:00:00 2001 From: Jim Burtoft <39492751+jimburtoft@users.noreply.github.com> Date: Wed, 17 Jul 2024 10:19:54 -0400 Subject: [PATCH] Fix spelling in error message I THINK that Talse should be False here. The boolean flips depending on whether we are looking at inline_weights_to_neff or is_weights_neff_separated. It looks like the only time you would get to this is if inline_weights_to_neff is set to True, so the recommendation would b to switch it to False. --- optimum/neuron/utils/misc.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/optimum/neuron/utils/misc.py b/optimum/neuron/utils/misc.py index c2c137898..a845ce43e 100644 --- a/optimum/neuron/utils/misc.py +++ b/optimum/neuron/utils/misc.py @@ -549,7 +549,7 @@ def replace_weights( prefix: str = "model", ): """ - Replaces the weights in a Neuron Model with weights from another model, the original neuron model should have separated weights(by setting `inline_weights_to_neff=Talse` during the tracing). + Replaces the weights in a Neuron Model with weights from another model, the original neuron model should have separated weights(by setting `inline_weights_to_neff=False` during the tracing). """ if isinstance(weights, torch.nn.Module): @@ -592,7 +592,7 @@ def _is_weights_neff_separated(config): if weights is not None and not is_weights_neff_separated: raise RuntimeError( - "Unable to replace weights of the neuron model since its weights and neff are not separated, please set `inline_weights_to_neff=Talse` when converting the model to Neuron format." + "Unable to replace weights of the neuron model since its weights and neff are not separated, please set `inline_weights_to_neff=False` when converting the model to Neuron format." )