Update src/deepsparse/evaluation/utils.py

neuralmagic · Mar 7, 2024 · ab18e23 · ab18e23
1 parent 83d8ac6
commit ab18e23
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/deepsparse/evaluation/utils.py b/src/deepsparse/evaluation/utils.py
@@ -199,7 +199,7 @@ def create_pipeline(
         text generation model from. This can be a local
         or remote path to the model or a sparsezoo stub
     :param engine_type: The engine type to initialize the model with.
-    :return: The initialized pipeline
+    :return: The initialized pipeline and the mutated (potentially reduced number of) kwargs
     """
     engine_type = engine_type or DEEPSPARSE_ENGINE
     return (