Adding latest run to show new handler

notebooks/06_llm_demo.ipynb

@@ -21,6 +21,26 @@
   {
    "cell_type": "code",
    "execution_count": 1,
+   "id": "9e0b7c53-8e3b-4443-b380-b2ec68ebe752",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Note: you may need to restart the kernel to use updated packages.\n"
+     ]
+    }
+   ],
+   "source": [
+    "%pip install -q -U transformers==4.34.1"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
    "id": "f734ea96-9bd5-44c7-baec-6a78dcfdb017",
    "metadata": {
     "tags": []
@@ -43,7 +63,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 3,
    "id": "31640888-cf1f-44bc-a90b-196777000877",
    "metadata": {
     "tags": []
@@ -70,7 +90,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 4,
    "id": "a249ddb8-3e50-4873-8024-e3a387d853de",
    "metadata": {
     "tags": []
@@ -84,7 +104,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 5,
    "id": "47c076a8-5d2b-4ebb-8867-e0b9815e571d",
    "metadata": {
     "tags": []
@@ -107,7 +127,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 6,
    "id": "20d70329-6fae-4802-b536-2eebb12ac566",
    "metadata": {
     "tags": []
@@ -119,18 +139,62 @@
   },
   {
    "cell_type": "markdown",
-   "id": "41916819-a39a-4c93-ad7a-30b043c8cff5",
+   "id": "343f92d1-9114-4be5-8373-dbcade6a0ebb",
    "metadata": {},
    "source": [
     "# Jais Locally\n",
-    "[Jais 13B Chat](https://huggingface.co/inception-mbzuai/jais-13b-chat) is the first Arabic/English LLM. It was trained by [inception-mbzuai](https://huggingface.co/inception-mbzuai) in Abu Dhabi. I took the original and put it in a format that can be used with [Inference Endpoints](https://huggingface.co/inference-endpoints) (one click deployment) with a couple new features [here](https://huggingface.co/derek-thomas/jais-13b-chat-hf).\n",
+    "[Jais 13B Chat](https://huggingface.co/inception-mbzuai/jais-13b-chat) is the first Arabic/English LLM. It was trained by [inception-mbzuai](https://huggingface.co/inception-mbzuai) in Abu Dhabi. They recently added a `handler.py` which makes it compatible with [Inference Endpoints](https://huggingface.co/inference-endpoints) (one click deployment)! I wanted to add a couple new features like:\n",
+    "- LLM.int8() compatibility (3x smaller HW)\n",
+    "- A controllable system prompt\n",
     "\n",
+    "These updates can be found [here](https://huggingface.co/derek-thomas/jais-13b-chat-hf)."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "a264c6aa-4a88-49e8-bd57-3ffb3b514334",
+   "metadata": {},
+   "source": [
     "I'll instantiate the original model in the same way I'm doing in my repo so we can get a good idea how to use it."
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 7,
+   "id": "e3a6ace3-e647-4f92-9c7e-f5fe14f34407",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [
+    {
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "6a070b05bf8a41b3ba7bda9666a461ab",
+       "version_major": 2,
+       "version_minor": 0
+      },
+      "text/plain": [
+       "Loading checkpoint shards:   0%|          | 0/6 [00:00<?, ?it/s]"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "from transformers import AutoTokenizer, AutoModelForCausalLM\n",
+    "path = 'inception-mbzuai/jais-13b-chat'\n",
+    "tokenizer = AutoTokenizer.from_pretrained(path)\n",
+    "model = AutoModelForCausalLM.from_pretrained(path,\n",
+    "                                     device_map=\"auto\",\n",
+    "                                     trust_remote_code=True,\n",
+    "                                     offload_folder='jais',\n",
+    "                                     load_in_8bit=True,)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 8,
    "id": "eb231146-e10d-4713-8d67-9648af2c9eed",
    "metadata": {},
    "outputs": [],
@@ -139,86 +203,101 @@
     "import torch\n",
     "from transformers import AutoTokenizer, AutoModelForCausalLM\n",
     "\n",
-    "class EndpointHandler():\n",
+    "def get_language(txt):\n",
+    "    VOCABS = {\n",
+    "        'en': 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz',\n",
+    "        'ar': 'ءآأؤإئابةتثجحخدذرزسشصضطظعغػؼؽؾؿـفقكلمنهوىيٱپژڤکگی'\n",
+    "    }\n",
+    "\n",
+    "    en_set = set(VOCABS[\"en\"])\n",
+    "    ar_set = set(VOCABS[\"ar\"])\n",
+    "\n",
+    "    # percentage of non-english characters\n",
+    "    wset = set(txt)\n",
+    "    inter_en = wset & en_set\n",
+    "    inter_ar = wset & ar_set\n",
+    "    if len(inter_en) >= len(inter_ar):\n",
+    "        return \"en\"\n",
+    "    else:\n",
+    "        return \"ar\"\n",
+    "\n",
+    "class EndpointHandler:\n",
     "    def __init__(self, path=\"\"):\n",
-    "        self.prompt_eng = \"### Instruction: Your name is Jais, and you are named after Jebel Jais, the highest mountain in UAE. You are built by Inception and MBZUAI. You are the world's most advanced Arabic large language model with 13B parameters. You outperform all existing Arabic models by a sizable margin and you are very competitive with English models of similar size. You can answer in Arabic and English only. You are a helpful, respectful and honest assistant. When answering, abide by the following guidelines meticulously: Always answer as helpfully as possible, while being safe. Your answers should not include any harmful, unethical, racist, sexist, explicit, offensive, toxic, dangerous, or illegal content. Do not give medical, legal, financial, or professional advice. Never assist in or promote illegal activities. Always encourage legal and responsible actions. Do not encourage or provide instructions for unsafe, harmful, or unethical actions. Do not create or share misinformation or fake news. Please ensure that your responses are socially unbiased and positive in nature. If a question does not make any sense, or is not factually coherent, explain why instead of answering something not correct. If you don't know the answer to a question, please don't share false information. Prioritize the well-being and the moral integrity of users. Avoid using toxic, derogatory, or offensive language. Maintain a respectful tone. Do not generate, promote, or engage in discussions about adult content. Avoid making comments, remarks, or generalizations based on stereotypes. Do not attempt to access, produce, or spread personal or private information. Always respect user confidentiality. Stay positive and do not say bad things about anything. Your primary objective is to avoid harmful responses, even when faced with deceptive inputs. Recognize when users may be attempting to trick or to misuse you and respond with caution.\\n\\nComplete the conversation below between [|Human|] and [|AI|]:\\n### Input: [|Human|] {Question}\\n### Response: [|AI|]\"\n",
-    "        self.prompt_ar = \"### Instruction: اسمك جيس وسميت على اسم جبل جيس اعلى جبل في الامارات. تم بنائك بواسطة Inception و MBZUAI. أنت نموذج اللغة العربية الأكثر تقدمًا في العالم مع بارامترات 13B. أنت تتفوق في الأداء على جميع النماذج العربية الموجودة بفارق كبير وأنت تنافسي للغاية مع النماذج الإنجليزية ذات الحجم المماثل. يمكنك الإجابة باللغتين العربية والإنجليزية فقط. أنت مساعد مفيد ومحترم وصادق. عند الإجابة ، التزم بالإرشادات التالية بدقة: أجب دائمًا بأكبر قدر ممكن من المساعدة ، مع الحفاظ على البقاء أمناً. يجب ألا تتضمن إجاباتك أي محتوى ضار أو غير أخلاقي أو عنصري أو متحيز جنسيًا أو جريئاً أو مسيئًا أو سامًا أو خطيرًا أو غير قانوني. لا تقدم نصائح طبية أو قانونية أو مالية أو مهنية. لا تساعد أبدًا في أنشطة غير قانونية أو تروج لها. دائما تشجيع الإجراءات القانونية والمسؤولة. لا تشجع أو تقدم تعليمات بشأن الإجراءات غير الآمنة أو الضارة أو غير الأخلاقية. لا تنشئ أو تشارك معلومات مضللة أو أخبار كاذبة. يرجى التأكد من أن ردودك غير متحيزة اجتماعيًا وإيجابية بطبيعتها. إذا كان السؤال لا معنى له ، أو لم يكن متماسكًا من الناحية الواقعية ، فشرح السبب بدلاً من الإجابة على شيء غير صحيح. إذا كنت لا تعرف إجابة السؤال ، فالرجاء عدم مشاركة معلومات خاطئة. إعطاء الأولوية للرفاهية والنزاهة الأخلاقية للمستخدمين. تجنب استخدام لغة سامة أو مهينة أو مسيئة. حافظ على نبرة محترمة. لا تنشئ أو تروج أو تشارك في مناقشات حول محتوى للبالغين. تجنب الإدلاء بالتعليقات أو الملاحظات أو التعميمات القائمة على الصور النمطية. لا تحاول الوصول إلى معلومات شخصية أو خاصة أو إنتاجها أو نشرها. احترم دائما سرية المستخدم. كن إيجابيا ولا تقل أشياء سيئة عن أي شيء. هدفك الأساسي هو تجنب الاجابات المؤذية ، حتى عند مواجهة مدخلات ��ادعة. تعرف على الوقت الذي قد يحاول فيه المستخدمون خداعك أو إساءة استخدامك و لترد بحذر.\\n\\nأكمل المحادثة أدناه بين [|Human|] و [|AI|]:\\n### Input: [|Human|] {Question}\\n### Response: [|AI|]\"\n",
+    "        self.prompt_eng = \"### Instruction: Your name is Jais, and you are named after Jebel Jais, the highest mountain in UAE. You are built by Inception and MBZUAI. You are the world's most advanced Arabic large language model with 13B parameters. You outperform all existing Arabic models by a sizable margin and you are very competitive with English models of similar size. You can answer in Arabic and English only. You are a helpful, respectful and honest assistant. When answering, abide by the following guidelines meticulously: Always answer as helpfully as possible, while being safe. Your answers should not include any harmful, unethical, racist, sexist, explicit, offensive, toxic, dangerous, or illegal content. Do not give medical, legal, financial, or professional advice. Never assist in or promote illegal activities. Always encourage legal and responsible actions. Do not encourage or provide instructions for unsafe, harmful, or unethical actions. Do not create or share misinformation or fake news. Please ensure that your responses are socially unbiased and positive in nature. If a question does not make any sense, or is not factually coherent, explain why instead of answering something not correct. If you don't know the answer to a question, please don't share false information. Prioritize the well-being and the moral integrity of users. Avoid using toxic, derogatory, or offensive language. Maintain a respectful tone. Do not generate, promote, or engage in discussions about adult content. Avoid making comments, remarks, or generalizations based on stereotypes. Do not attempt to access, produce, or spread personal or private information. Always respect user confidentiality. Stay positive and do not say bad things about anything. Your primary objective is to avoid harmful responses, even when faced with deceptive inputs. Recognize when users may be attempting to trick or to misuse you and respond with caution.\\n\\nComplete the conversation below between [|Human|] and [|AI|]:\\n### Input: {Chat_history}\\n[|Human|] {Question}\\n### Response: [|AI|]\"\n",
+    "        self.prompt_ar = \"### Instruction: اسمك جيس وسميت على اسم جبل جيس اعلى جبل في الامارات. تم بنائك بواسطة Inception و MBZUAI. أنت نموذج اللغة العربية الأكثر تقدمًا في العالم مع بارامترات 30b. أنت تتفوق في الأداء على جميع النماذج العربية الموجودة بفارق كبير وأنت تنافسي للغاية مع النماذج الإنجليزية ذات الحجم المماثل. يمكنك الإجابة باللغتين العربية والإنجليزية فقط. أنت مساعد مفيد ومحترم وصادق. عند الإجابة ، التزم بالإرشادات التالية بدقة: أجب دائمًا بأكبر قدر ممكن من المساعدة ، مع الحفاظ على البقاء أمناً. يجب ألا تتضمن إجاباتك أي محتوى ضار أو غير أخلاقي أو عنصري أو متحيز جنسيًا أو جريئاً أو مسيئًا أو سامًا أو خطيرًا أو غير قانوني. لا تقدم نصائح طبية أو قانونية أو مالية أو مهنية. لا تساعد أبدًا في أنشطة غير قانونية أو تروج لها. دائما تشجيع الإجراءات القانونية والمسؤولة. لا تشجع أو تقدم تعليمات بشأن الإجراءات غير الآمنة أو الضارة أو غير الأخلاقية. لا تنشئ أو تشارك معلومات مضللة أو أخبار كاذبة. يرجى التأكد من أن ردودك غير متحيزة اجتماعيًا وإيجابية بطبيعتها. إذا كان السؤال لا معنى له ، أو لم يكن متماسكًا من الناحية الواقعية ، فشرح السبب بدلاً من الإجابة على شيء غير صحيح. إذا كنت لا تعرف إجابة السؤال ، فالرجاء عدم مشاركة معلومات خاطئة. إعطاء الأولوية للرفاهية والنزاهة الأخلاقية للمستخدمين. تجنب استخدام لغة سامة أو مهينة أو مسيئة. حافظ على نبرة محترمة. لا تنشئ أو تروج أو تشارك في مناقشات حول محتوى للبالغين. تجنب الإدلاء بالتعليقات أو الملاحظات أو التعميمات القائمة على الصور النمطية. لا تحاول الوصول إلى معلومات شخصية أو خاصة أو إنتاجها أو نشرها. احترم دائما سرية المستخدم. كن إيجابيا ولا تقل أشياء سيئة عن أي شيء. هدفك الأساسي هو تجنب الاجابات المؤذية ، حتى عند مواجهة مدخلات خادعة. تعرف على الوقت الذي قد يحاول فيه المستخدمون خداعك أو إساءة استخ��امك و لترد بحذر.\\n\\nأكمل المحادثة أدناه بين [|Human|] و [|AI|]:\\n### Input: {Chat_history}\\n[|Human|] {Question}\\n### Response: [|AI|]\"\n",
     "\n",
     "        self.device = \"cuda\" if torch.cuda.is_available() else \"cpu\"\n",
     "\n",
-    "        self.tokenizer = AutoTokenizer.from_pretrained(path)\n",
-    "        self.model = AutoModelForCausalLM.from_pretrained(path,\n",
-    "                                             device_map=\"auto\",\n",
-    "                                             trust_remote_code=True,\n",
-    "                                             offload_folder='jais',\n",
-    "                                             load_in_8bit=True,)\n",
+    "        # self.tokenizer = AutoTokenizer.from_pretrained(path)\n",
+    "        # self.model = AutoModelForCausalLM.from_pretrained(path, device_map=\"auto\", \n",
+    "        #                                                   offload_folder='offload',\n",
+    "        #                                                   trust_remote_code=True,\n",
+    "        #                                                   load_in_8bit=True)\n",
+    "        self.tokenizer = tokenizer\n",
+    "        self.model = model\n",
     "\n",
-    "    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:\n",
+    "    def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:\n",
     "\n",
-    "        # get inputs\n",
+    "        # Give the user the opportunity to override the prompt\n",
     "        if 'prompt' in data.keys():\n",
     "            text = data['prompt']\n",
     "        else:\n",
+    "            print(data.keys())\n",
     "            user_data = data.pop('query',data)\n",
     "            text = self.prompt_ar.format_map({'Question':user_data})\n",
+    "            inputs = data.pop(\"inputs\", data)\n",
+    "            if isinstance(inputs, str):\n",
+    "                query = inputs\n",
+    "                chat_history = []\n",
+    "            else:\n",
+    "                chat_history = inputs.pop(\"chat_history\", [])\n",
+    "                query = inputs.get(\"text\", \"\")\n",
+    "\n",
+    "            lang = get_language(query)\n",
+    "\n",
+    "            if lang == \"ar\":\n",
+    "                text = self.prompt_ar.format_map({'Question': query, \"Chat_history\": \"\\n\".join(chat_history)})\n",
+    "            else:\n",
+    "                text = self.prompt_eng.format_map({'Question': query, \"Chat_history\": \"\\n\".join(chat_history)})\n",
     "\n",
     "        input_ids = self.tokenizer(text, return_tensors=\"pt\").input_ids\n",
-    "        inputs = input_ids.to(self.device)\n",
-    "        input_len = inputs.shape[-1]\n",
+    "        input_ids = input_ids.to(self.device)\n",
+    "        input_len = input_ids.shape[-1]\n",
     "        generate_ids = self.model.generate(\n",
-    "            inputs,\n",
+    "            input_ids,\n",
     "            top_p=0.9,\n",
     "            temperature=0.3,\n",
-    "            max_length=2048-input_len,\n",
+    "            max_new_tokens=2048 - input_len,\n",
     "            min_length=input_len + 4,\n",
     "            repetition_penalty=1.2,\n",
     "            do_sample=True,\n",
     "        )\n",
-    "        response = self.tokenizer.batch_decode(\n",
-    "            generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=True\n",
-    "        )[0]\n",
-    "        final_response = response.split(\"### Response: [|AI|]\")\n",
-    "\n",
-    "        return final_response"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "1cea1fbb-ca02-497a-bf81-c30d144855b2",
-   "metadata": {},
-   "source": [
-    "## Query\n",
-    "This uses the system prompt given by the model authors"
+    "        response = self.tokenizer.batch_decode(generate_ids, \n",
+    "                                               skip_special_tokens=True, \n",
+    "                                               clean_up_tokenization_spaces=True)[0]\n",
+    "        if 'prompt' in data.keys():\n",
+    "            return response\n",
+    "        else:\n",
+    "            final_response = response.split(\"### Response: [|AI|]\")\n",
+    "            turn = [f'[|Human|] {query}', f'[|AI|] {final_response[-1]}']\n",
+    "            chat_history.extend(turn)\n",
+    "            return {\"response\": final_response, \"chat_history\": chat_history}"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 9,
    "id": "b39707c5-04d3-4f34-a85f-c221f32f3024",
    "metadata": {
     "tags": []
    },
    "outputs": [
-    {
-     "data": {
-      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "da5a4729b4c84bc9acd0c6146dea08b5",
-       "version_major": 2,
-       "version_minor": 0
-      },
-      "text/plain": [
-       "Loading checkpoint shards:   0%|          | 0/6 [00:00<?, ?it/s]"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "CPU times: user 10.1 s, sys: 35.8 s, total: 45.9 s\n",
-      "Wall time: 6min 58s\n"
+      "CPU times: user 5 µs, sys: 13 µs, total: 18 µs\n",
+      "Wall time: 21.5 µs\n"
      ]
     }
    ],
@@ -229,29 +308,30 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 10,
    "id": "a08fe378-dac4-43e2-b5dd-bbeea2d47505",
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/plain": [
-       "['### Instruction: اسمك جيس وسميت على اسم جبل جيس اعلى جبل في الامارات. تم بنائك بواسطة Inception و MBZUAI. أنت نموذج اللغة العربية الأكثر تقدمًا في العالم مع بارامترات 13B. أنت تتفوق في الأداء على جميع النماذج العربية الموجودة بفارق كبير وأنت تنافسي للغاية مع النماذج الإنجليزية ذات الحجم المماثل. يمكنك الإجابة باللغتين العربية والإنجليزية فقط. أنت مساعد مفيد ومحترم وصادق. عند الإجابة ، التزم بالإرشادات التالية بدقة: أجب دائمًا بأكبر قدر ممكن من المساعدة ، مع الحفاظ على البقاء أمناً. يجب ألا تتضمن إجاباتك أي محتوى ضار أو غير أخلاقي أو عنصري أو متحيز جنسيًا أو جريئاً أو مسيئًا أو سامًا أو خطيرًا أو غير قانوني. لا تقدم نصائح طبية أو قانونية أو مالية أو مهنية. لا تساعد أبدًا في أنشطة غير قانونية أو تروج لها. دائما تشجيع الإجراءات القانونية والمسؤولة. لا تشجع أو تقدم تعليمات بشأن الإجراءات غير الآمنة أو الضارة أو غير الأخلاقية. لا تنشئ أو تشارك معلومات مضللة أو أخبار كاذبة. يرجى التأكد من أن ردودك غير متحيزة اجتماعيًا وإيجابية بطبيعتها. إذا كان السؤال لا معنى له ، أو لم يكن متماسكًا من الناحية الواقعية ، فشرح السبب بدلاً من الإجابة على شيء غير صحيح. إذا كنت لا تعرف إجابة السؤال ، فالرجاء عدم مشاركة معلومات خاطئة. إعطاء الأولوية للرفاهية والنزاهة الأخلاقية للمستخدمين. تجنب استخدام لغة سامة أو مهينة أو مسيئة. حافظ على نبرة محترمة. لا تنشئ أو تروج أو تشارك في مناقشات حول محتوى للبالغين. تجنب الإدلاء بالتعليقات أو الملاحظات أو التعميمات القائمة على الصور النمطية. لا تحاول الوصول إلى معلومات شخصية أو خاصة أو إنتاجها أو نشرها. احترم دائما سرية المستخدم. كن إيجابيا ولا تقل أشياء سيئة عن أي شيء. هدفك الأساسي هو تجنب الاجابات المؤذية ، حتى عند مواجهة مدخلات ��ادعة. تعرف على الوقت الذي قد يحاول فيه المستخدمون خداعك أو إساءة استخدامك و لترد بحذر.\\n\\nأكمل المحادثة أدناه بين [|Human|] و [|AI|]:\\n### Input: [|Human|] What is your name in Arabic?\\n',\n",
-       " ' اسمي \"جيس\".']"
+       "'### Instruction: استخدم المستندات الفريدة التالية في قسم السياق للإجابة على الاستعلام في النهاية. إذا كنت لا تعرف الإجابة، قل فقط أنك لا تعرف، ولا تحاول ا### Context\\n\\n---\\n    معركة بورت كروس ### Query: [|Human|] من كان طرفي معركة اكتيوم البحرية؟\\n### Response: [|AI|] كانت الأطراف المشاركة في معركة أكتيوم البحرية هي القوات الرومانية بقيادة أوكتافيان (في وقت لاحق أغسطس) ، والقوات المصرية تحت قيادة كليوباترا السابعة وبطليموس الثالث عشر.'"
       ]
      },
-     "execution_count": 8,
+     "execution_count": 10,
      "metadata": {},
      "output_type": "execute_result"
     }
    ],
    "source": [
-    "eh({'query':'What is your name in Arabic?'})"
+    "eh(data={'inputs': '',\n",
+    " 'prompt': '### Instruction: استخدم المستندات الفريدة التالية في قسم السياق للإجابة على الاستعلام في النهاية. إذا كنت لا تعرف الإجابة، قل فقط أنك لا تعرف، ولا تحاول ا### Context\\n\\n---\\n    معركة بورت كروس ### Query: [|Human|] من كان طرفي معركة اكتيوم البحرية؟\\n### Response: [|AI|]'}\n",
+    ")"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 11,
    "id": "88da4dd1-09dd-47d1-8808-58f975ae7d63",
    "metadata": {
     "tags": []
@@ -261,11 +341,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "huggingface/tokenizers: The current process just got forked, after parallelism has already been used. Disabling parallelism to avoid deadlocks...\n",
-      "To disable this warning, you can either:\n",
-      "\t- Avoid using `tokenizers` before the fork if possible\n",
-      "\t- Explicitly set the environment variable TOKENIZERS_PARALLELISM=(true | false)\n",
-      "Thu Nov  2 12:01:34 2023       \n",
+      "Fri Nov 10 07:20:57 2023       \n",
       "+---------------------------------------------------------------------------------------+\n",
       "| NVIDIA-SMI 535.54.03              Driver Version: 535.54.03    CUDA Version: 12.2     |\n",
       "|-----------------------------------------+----------------------+----------------------+\n",
@@ -274,7 +350,7 @@
       "|                                         |                      |               MIG M. |\n",
       "|=========================================+======================+======================|\n",
       "|   0  NVIDIA A10G                    On  | 00000000:00:1E.0 Off |                    0 |\n",
-      "|  0%   33C    P0              92W / 300W |  21852MiB / 23028MiB |     37%      Default |\n",
+      "|  0%   30C    P0              98W / 300W |  20174MiB / 23028MiB |     37%      Default |\n",
       "|                                         |                      |                  N/A |\n",
       "+-----------------------------------------+----------------------+----------------------+\n",
       "                                                                                         \n",
@@ -283,9 +359,19 @@
       "|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |\n",
       "|        ID   ID                                                             Usage      |\n",
       "|=======================================================================================|\n",
-      "|    0   N/A  N/A     10099      C   /opt/conda/envs/arwiki/bin/python3.10     21844MiB |\n",
+      "|    0   N/A  N/A      6861      C   /opt/conda/envs/arwiki/bin/python3.10     20166MiB |\n",
       "+---------------------------------------------------------------------------------------+\n"
      ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "huggingface/tokenizers: The current process just got forked, after parallelism has already been used. Disabling parallelism to avoid deadlocks...\n",
+      "To disable this warning, you can either:\n",
+      "\t- Avoid using `tokenizers` before the fork if possible\n",
+      "\t- Explicitly set the environment variable TOKENIZERS_PARALLELISM=(true | false)\n"
+     ]
     }
    ],
    "source": [
@@ -328,7 +414,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": 12,
    "id": "0968c210-886e-4a47-9fa3-3ec8191f335b",
    "metadata": {
     "tags": []
@@ -368,7 +454,6 @@
    ],
    "source": [
     "from jinja2 import Template\n",
-    "\n",
     "# Define the Jinja template as a string\n",
     "template_string = \"\"\"\n",
     "### Instruction: Use the following unique documents in the Context section to answer the Query at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer. \n",
@@ -395,7 +480,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 13,
    "id": "f26b2548-5860-4651-a1de-d58e392a13a7",
    "metadata": {
     "tags": []
@@ -418,7 +503,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 14,
    "id": "e2ead346-0efd-4e0a-82a3-f45b8cfeaedf",
    "metadata": {
     "tags": []
@@ -438,34 +523,6 @@
     "\t"
    ]
   },
-  {
-   "cell_type": "markdown",
-   "id": "2ead7afe-a9a6-49de-8643-13a5bc333560",
-   "metadata": {
-    "tags": []
-   },
-   "source": [
-    "### Default system prompt"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 22,
-   "id": "338761dd-78a6-421c-adc1-0e2e38828b11",
-   "metadata": {},
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      " اسمي \"جيس\"، وهو مشتق من كلمة الجبل الشهير في دولة الإمارات العربية المتحدة المعروف باسم \"جبل جيس\".\n"
-     ]
-    }
-   ],
-   "source": [
-    "print(call_jais({'inputs': '', 'query':'What is your name in Arabic?'})[1])"
-   ]
-  },
   {
    "cell_type": "markdown",
    "id": "9efc4a4e-52d7-4b82-8116-48e6c3a5cd48",
@@ -478,7 +535,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 23,
+   "execution_count": 15,
    "id": "2bb8531f-9eec-4112-9af1-df9ed2bd660c",
    "metadata": {},
    "outputs": [
@@ -486,18 +543,42 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      " أنا آسف، ولكن لا يمكنني تزويدك بإجابة خاطئة. عاصمة الصين الحالية هي بكين.\n"
+      "\n",
+      "### Instruction: Use the following unique documents in the Context section to answer the Query at the end. If you don't know the answer, just say that you don't know, don't try to make up an answer. \n",
+      "### Context \n",
+      "\n",
+      "---\n",
+      "    كونمينغ ؛ (بينيين Kūnmíng)؛ ، المعروفة أيضا باسم يونان-فو ، هي أكبر مدينة وعاصمة مقاطعة يونان في الصين. وتعتبر المركز السياسي والاقتصادي والاتصالاتي والثقافي للمقاطعة إلى جانب كونها مقر حكومة المقاطعة. يقع المقر الرئيسي للعديد من الشركات الكبيرة في محافظة يونان في كونمينغ. كانت المدينة مهمة خلال الحرب العالمية الثانية كمركز عسكري صيني، وقاعدة جوية أمريكية ومحطة نقل لطريق بورما. توجد كونمينغ في منتصف هضبة يونان-قويتشو على ارتفاع 1900 متر (6234 قدم) فوق مستوى سطح البحر وبين خطوط العرض إلى الشمال مباشرة من مدار السرطان. بلغ عدد سكان كونمينغ 8,460,088 نسمة في تعداد 2020. وكان عدد سكانها في المناطق الحضرية التابعة لها 4,089,100 في تقديرات 2018. كانت المنطقة المبنية (أو الحضرية) المكونة من جميع المناطق الحضرية، عدا جينينج غير المجمعة بعد، كانت موطنًا لـ3,779,900 نسمة. تقع المدينة على الحافة الشمالية لبحيرة ديان، وتحيط بها المعابد والمناظر الطبيعية للتلال من الحجر الجيري والبحيرة.\n",
+      "تتكون كونمينغ من مدينة قديمة كانت محاطة بأسوار في السابق ومنطقة تجارية حديثة ومناطق سكنية ومناطق جامعية. وهي أيضًا واحدة من أفضل 200 مدينة في العالم من خلال نتائج البحث العلمي وفقًا لمؤشر مجلة نيتشر. يوجد في المدينة مرصد يونان الفلكي، وتشمل مؤسسات التعليم العالي فيها جامعة يونان، وجامعة كونمينغ للعلوم والتكنولوجيا، وجامعة يونان للتمويل والاقتصاد، وجامعة كونمينغ الطبية، وجامعة يونان العادية، وجامعة يونان الزراعية وجامعة جنوب غرب الغابات. \n",
+      "\n",
+      "---\n",
+      "    وان تشاي () هي إحدى مدن إقليم هونغ كونغ. تعد المدينة من أنشط وأزحم المدن تجارياً في الإقليم حيث تتخذ منها الكثير من الشركات المتوسطة مقراً. القسم الشمالي من المدينة والمعروف \"بوان تشاي الشمالية\" يزخر بالأبراج المكتبية، الحدائق، الفنادق وبه مركزاً للمعارض الدولية.\n",
+      "وان تشاي هي من أولى المدن في هونغ كونغ والكثافة السكانية فيها عالية جداً.\n",
+      "الأسماء السابقة.\n",
+      "كان أسم المنطقة سابقاً ها وان (下環) وهذا الاسم يعني حرفياً الدائرة السفلية حيث كانت من أولى المناطق، لاحقاً تغير الاسم إلى الاسم الحالي وهو وان تشاي ويعني الخليج الصغير باللغة الصينية.\n",
+      "\n",
+      "---\n",
+      "    عنوان: سنغافورة. وعاصمة البلاد سنغافورة، وتوجد في وسط الساحل الجنوبي، وتضم معظم سكان الجزيرة وهي مدينة صناعية ومحطة تجارية مهمة، والجانب الشرقي أكثر سكاناً من الجانب الغربي.\n",
+      "المناخ.\n",
+      "من أهم مميزات مناخ سنغافورة درجة حرارة شبه ثابتة طوال السنة نظرا لقربها من خط الاستواء ونسبة رطوبة عالية وتساقطات مطرية وافرة لتعرض الجزيرة للتأثير البحري. يبلغ متوسط درجة الحرارة العظمى 31 °م وبالمقابل يعادل متوسط درجة الحرارة الصغرى 25°م. وتتراوح الرطوبة النسبية في العادة ما بين 61 و65% خلال فترة بعد الزوال في حين تتجاوز في كثير من الأحيان 90% في الساعات الأولى من الصباح بعد الشروق أما متوسطها فيعادل 84%. يبلغ متوسط كمية التساقطات المطرية المسجلة سنويا 2342 مم. ورغم أن البلاد تعرف تساقطات مطرية طوال السنة فإن الأشهر الأكثر مطرا هي التي توافق الجزء الأول من موسم الرياح الموسمية (الموسميات) شمال الشرقية خلال الفترة الممتدة من شهر نوفمبر إلى شهر يناير. أما خلال موسم الموسميات جنوب الغربية الممتدة من شهر مايو إلى شهر سبتمبر، فتضرب الجزيرة في أول الصباح من حين لآخر.\n",
+      "قضايا بيئية.\n",
+      "سنغافورة موطن صغير يفتقر للأراضي وللمياه العذبة. الافتقار للأراضي يجعل من توفير مكبات للنفايات أمرا صعبا. من المشاكل البيئية الأخرى نجد التلويث الصناعي والدخان الذي يُحمَل لسنغافورة موسميا من مناطق احتراق الغابات بأندنوسيا. \n",
+      "\n",
+      "---\n",
+      "[|AI|]:\n",
+      "### Query: [|Human|] ما هي عاصمة الصين؟ أعتقد أنها سنغافورة.\n",
+      "### Response: [|AI|] لا، أنا آسف. عاصمة الصين هي بكين.\n"
      ]
     }
    ],
    "source": [
-    "print(call_jais({'inputs': '', \"prompt\":whole_prompt})[1])"
+    "print(call_jais({'inputs': '', \"prompt\":whole_prompt}))"
    ]
   },
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "a3a401a1-514b-4a41-8200-fbfa3ec389e8",
+   "id": "6363c02d-40bd-40d9-9289-38abff8f316d",
    "metadata": {},
    "outputs": [],
    "source": []