diff --git a/optimum/neuron/modeling_diffusion.py b/optimum/neuron/modeling_diffusion.py
index 6a85e738f..f59f8f9a8 100644
--- a/optimum/neuron/modeling_diffusion.py
+++ b/optimum/neuron/modeling_diffusion.py
@@ -823,7 +823,6 @@ def _export(
                 lora_adapter_names=lora_adapter_names,
                 lora_scales=lora_scales,
                 library_name=cls.library_name,
-                checked_config=cache_config,
                 **input_shapes,
             )
 
diff --git a/tests/exporters/exporters_utils.py b/tests/exporters/exporters_utils.py
index 484f41c14..7dd5303b5 100644
--- a/tests/exporters/exporters_utils.py
+++ b/tests/exporters/exporters_utils.py
@@ -71,6 +71,6 @@
     "image_batch_size": 1,
 }
 
-WEIGHTS_NEFF_SEPARATION_UNSUPPORTED_ARCH = ["camembert", "roberta"]
+WEIGHTS_NEFF_SEPARATION_UNSUPPORTED_ARCH = ["camembert", "roberta", "mobilenet-v2"]
 
 SEED = 42