Update notebook.ipynb

notebook.ipynb

@@ -68,7 +68,7 @@
       "cell_type": "code",
       "source": [
         "from litert_tools.pipeline import pipeline\n",
-        "runner = pipeline.load(\"litert-community/Qwen2.5-1.5B-Instruct\", \"Qwen2.5-1.5B-Instruct_seq128_q8_ekv1280.task\")"
+        "runner = pipeline.load(\"Qwen2.5-1.5B-Instruct_seq128_q8_ekv1280.task\", repo_id=\"litert-community/Qwen2.5-1.5B-Instruct\")"
       ],
       "metadata": {
         "id": "3t47HAG2tvc3"
@@ -90,7 +90,7 @@
       "source": [
         "# Disclaimer: Model performance demonstrated with the Python API in this notebook is not representative of performance on a local device.\n",
         "prompt = \"What is the capital of France?\"\n",
-        "output = runner.generate(prompt, max_decode_steps=None)"
+        "output = runner.generate(prompt)"
       ],
       "metadata": {
         "id": "wT9BIiATkjzL"