fix(decoder): specify libraryname to suppress warning (#570)

huggingface · Apr 16, 2024 · 516c511 · 516c511
1 parent c3daf50
commit 516c511
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/optimum/neuron/modeling_decoder.py b/optimum/neuron/modeling_decoder.py
@@ -52,7 +52,9 @@
 
 
 def get_exporter(config, task):
-    return TasksManager.get_exporter_config_constructor(model_type=config.model_type, exporter="neuron", task=task)()
+    return TasksManager.get_exporter_config_constructor(
+        model_type=config.model_type, exporter="neuron", task=task, library_name="transformers"
+    )()
 
 
 # Note: with python 3.9, functools.cache would be more suited