From e1528a6e5451c4a050b74adb1ab7a41e5a5ac543 Mon Sep 17 00:00:00 2001 From: guylei-code Date: Tue, 5 May 2026 15:49:42 +0300 Subject: [PATCH] First commit, update changes requested in ipynb (#980) --- steps/src/toxicity_guardrail/toxicity_guardrail.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/steps/src/toxicity_guardrail/toxicity_guardrail.ipynb b/steps/src/toxicity_guardrail/toxicity_guardrail.ipynb index ddc3fe99..be9d6725 100644 --- a/steps/src/toxicity_guardrail/toxicity_guardrail.ipynb +++ b/steps/src/toxicity_guardrail/toxicity_guardrail.ipynb @@ -200,7 +200,7 @@ ")\n", "\n", "model_runner = ModelRunnerStep(name=\"llm_runner\")\n", - "model_runner.add_model(endpoint_name=\"llm_model\", model_class=\"LLMModel\")\n", + "model_runner.add_model(endpoint_name=\"llm_model\", model_class=\"LLMModel\", execution_mechanism=\"naive\")\n", "graph.add_step(model_runner, after=\"toxicity_guardrail\")\n", "\n", "graph.add_step(name=\"format_answer\", handler=\"format_answer\", after=\"llm_runner\").respond()\n",