notebook for RAG using gemma, huggingface and elastic

elastic · Mar 5, 2024 · a108f2a · a108f2a
1 parent df2fdd1
commit a108f2a
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/notebooks/integrations/gemma/rag-gemma-huggingface-elastic.ipynb b/notebooks/integrations/gemma/rag-gemma-huggingface-elastic.ipynb
@@ -249,15 +249,15 @@
    "outputs": [],
    "source": [
     "pipe = pipeline(\n",
-    " \"text-generation\",\n",
-    " model=model,\n",
-    " tokenizer=tokenizer,\n",
-    " max_new_tokens=1024,\n",
+    "    \"text-generation\",\n",
+    "    model=model,\n",
+    "    tokenizer=tokenizer,\n",
+    "    max_new_tokens=1024,\n",
     ")\n",
     "\n",
     "llm = HuggingFacePipeline(\n",
-    " pipeline=pipe,\n",
-    " model_kwargs={\"temperature\": 0.7},\n",
+    "    pipeline=pipe,\n",
+    "    model_kwargs={\"temperature\": 0.7},\n",
     ")"
    ]
   },