add diffusion

app.py

@@ -22,6 +22,9 @@ from optimum_benchmark.backends.openvino.utils import (
 from optimum_benchmark.backends.transformers_utils import (
     TASKS_TO_AUTO_MODEL_CLASS_NAMES,
 )
+from optimum_benchmark.backends.diffusers_utils import (
+    TASKS_TO_AUTO_PIPELINE_CLASS_NAMES,
+)
 from optimum_benchmark import (
     Benchmark,
     BenchmarkConfig,
@@ -41,7 +44,7 @@ BACKENDS = ["pytorch", "openvino"]
 BENCHMARKS_HF_TOKEN = os.getenv("BENCHMARKS_HF_TOKEN")
 BENCHMARKS_REPO_ID = "optimum-benchmark/OpenVINO-Benchmarks"
 TASKS = set(TASKS_TO_OVMODELS.keys() | TASKS_TO_OVPIPELINES) & set(
-    TASKS_TO_AUTO_MODEL_CLASS_NAMES.keys()
+    TASKS_TO_AUTO_MODEL_CLASS_NAMES.keys() | TASKS_TO_AUTO_PIPELINE_CLASS_NAMES.keys()
 )
 
 

config_store.py

@@ -60,7 +60,12 @@ def get_inference_config():
         "inference.generate_kwargs": gr.Textbox(
             label="inference.generate_kwargs",
             value="{'max_new_tokens': 32, 'min_new_tokens': 32}",
-            info="Additional python dict of kwargs to pass to the generate function",
+            info="Additional python dict of kwargs to pass to the generate method",
+        ),
+        "inference.call_kwargs": gr.Textbox(
+            label="inference.call_kwargs",
+            value="{'num_inference_steps': 4}",
+            info="Additional python dict of kwargs to pass to the __call__ method",
         ),
     }