Remove a line non-intentionally merged (#628)

* fix * skip a test for inf1
huggingface · Jun 12, 2024 · e15c2ce · e15c2ce
1 parent da9d261
commit e15c2ce
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git 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
@@ -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