revert the flag

huggingface · Jan 15, 2025 · 581e13d · 581e13d
1 parent a271839
commit 581e13d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/optimum/exporters/neuron/convert.py b/optimum/exporters/neuron/convert.py
@@ -557,7 +557,8 @@ def export_neuronx(
     compiler_args.extend(["--optlevel", optlevel])
     logger.info(f"Using Neuron: --optlevel {optlevel}")
 
-    compiler_args.extend(["--model-type", "transformer"])
+    if getattr(config._config, "is_encoder_decoder", False):
+        compiler_args.extend(["--model-type", "transformer"])
 
     compiler_args = add_stable_diffusion_compiler_args(config, compiler_args)  # diffusers specific