Skip to content

Commit ed14c68

Browse files
First commit, update changes requested in ipynb (#980) (#981)
Co-authored-by: guylei-code <guyleibu@gmail.com>
1 parent 2abb601 commit ed14c68

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

steps/src/toxicity_guardrail/toxicity_guardrail.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,7 @@
200200
")\n",
201201
"\n",
202202
"model_runner = ModelRunnerStep(name=\"llm_runner\")\n",
203-
"model_runner.add_model(endpoint_name=\"llm_model\", model_class=\"LLMModel\")\n",
203+
"model_runner.add_model(endpoint_name=\"llm_model\", model_class=\"LLMModel\", execution_mechanism=\"naive\")\n",
204204
"graph.add_step(model_runner, after=\"toxicity_guardrail\")\n",
205205
"\n",
206206
"graph.add_step(name=\"format_answer\", handler=\"format_answer\", after=\"llm_runner\").respond()\n",

0 commit comments

Comments
 (0)