use small model in llm agent notebook (#2132)

openvinotoolkit · Jun 19, 2024 · 1f11b58 · 1f11b58
1 parent 5cde8a3
commit 1f11b58
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 5 deletions.
diff --git a/.ci/ignore_treon_docker.txt b/.ci/ignore_treon_docker.txt
@@ -61,7 +61,6 @@ notebooks/animate-anyone/animate-anyone.ipynb
 notebooks/llava-next-multimodal-chatbot/llava-next-multimodal-chatbot.ipynb
 notebooks/llm-rag-langchain/llm-rag-langchain.ipynb
 notebooks/stable-video-diffusion/stable-video-diffusion.ipynb
-notebooks/llm-agent-langchain/llm-agent-langchain.ipynb
 notebooks/hello-npu/hello-npu.ipynb
 notebooks/stable-cascade-image-generation/stable-cascade-image-generation.ipynb
 notebooks/dynamicrafter-animating-images/dynamicrafter-animating-images.ipynb

diff --git a/.ci/ignore_treon_linux.txt b/.ci/ignore_treon_linux.txt
@@ -62,7 +62,6 @@ notebooks/animate-anyone/animate-anyone.ipynb
 notebooks/llava-next-multimodal-chatbot/llava-next-multimodal-chatbot.ipynb
 notebooks/llm-rag-langchain/llm-rag-langchain.ipynb
 notebooks/stable-video-diffusion/stable-video-diffusion.ipynb
-notebooks/llm-agent-langchain/llm-agent-langchain.ipynb
 notebooks/hello-npu/hello-npu.ipynb
 notebooks/stable-cascade-image-generation/stable-cascade-image-generation.ipynb
 notebooks/dynamicrafter-animating-images/dynamicrafter-animating-images.ipynb

diff --git a/.ci/ignore_treon_mac.txt b/.ci/ignore_treon_mac.txt
@@ -63,7 +63,6 @@ notebooks/animate-anyone/animate-anyone.ipynb
 notebooks/llava-next-multimodal-chatbot/llava-next-multimodal-chatbot.ipynb
 notebooks/llm-rag-langchain/llm-rag-langchain.ipynb
 notebooks/stable-video-diffusion/stable-video-diffusion.ipynb
-notebooks/llm-agent-langchain/llm-agent-langchain.ipynb
 notebooks/hello-npu/hello-npu.ipynb
 notebooks/stable-cascade-image-generation/stable-cascade-image-generation.ipynb
 notebooks/dynamicrafter-animating-images/dynamicrafter-animating-images.ipynb

diff --git a/.ci/ignore_treon_win.txt b/.ci/ignore_treon_win.txt
@@ -60,7 +60,6 @@ notebooks/animate-anyone/animate-anyone.ipynb
 notebooks/llava-next-multimodal-chatbot/llava-next-multimodal-chatbot.ipynb
 notebooks/llm-rag-langchain/llm-rag-langchain.ipynb
 notebooks/stable-video-diffusion/stable-video-diffusion.ipynb
-notebooks/llm-agent-langchain/llm-agent-langchain.ipynb
 notebooks/hello-npu/hello-npu.ipynb
 notebooks/stable-cascade-image-generation/stable-cascade-image-generation.ipynb
 notebooks/dynamicrafter-animating-images/dynamicrafter-animating-images.ipynb

diff --git a/notebooks/llm-agent-langchain/llm-agent-langchain.ipynb b/notebooks/llm-agent-langchain/llm-agent-langchain.ipynb
@@ -272,7 +272,10 @@
    "cell_type": "code",
    "execution_count": 5,
    "id": "981df8fe-cfcf-455a-919e-dda36f3b5dfb",
-   "metadata": {},
+   "metadata": {
+   "test_replace": {"mistralai/Mistral-7B-Instruct-v0.3": "TinyLlama/TinyLlama-1.1B-Chat-v1.0"}
+
+   },
    "outputs": [],
    "source": [
     "from pathlib import Path\n",