From 558379f8a1f7f67820ea219323f9af434a6ae2ba Mon Sep 17 00:00:00 2001 From: David Corvoysier Date: Thu, 6 Feb 2025 11:06:59 +0000 Subject: [PATCH] review: avoid error if transformers-neuronx is not installed --- optimum/exporters/neuron/__main__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/optimum/exporters/neuron/__main__.py b/optimum/exporters/neuron/__main__.py index 2b7e2e80a..43d9f6626 100644 --- a/optimum/exporters/neuron/__main__.py +++ b/optimum/exporters/neuron/__main__.py @@ -730,7 +730,7 @@ def main(): submodels = None else: input_shapes, neuron_config_class = get_input_shapes_and_config_class(task, args) - if NeuronDecoderExportConfig in inspect.getmro(neuron_config_class): + if is_transformers_neuronx_available() and NeuronDecoderExportConfig in inspect.getmro(neuron_config_class): # TODO: warn about ignored args: # dynamic_batch_size, compiler_workdir, optlevel, # atol, disable_validation, library_name