RAG_Mistral_7b.ipynb

{
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/Vishvam10/LLM_RAG/blob/master/RAG_Mistral_7b.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "KdoxDHTVgyZq"
      },
      "source": [
        "## **Imports and Installs**"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 1,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "ahg2XhwXPAdU",
        "outputId": "7e06de6f-af64-464a-da2c-6f0cedb8e4fe"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m755.5/755.5 MB\u001b[0m \u001b[31m1.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m507.1/507.1 kB\u001b[0m \u001b[31m36.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m8.4/8.4 MB\u001b[0m \u001b[31m59.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m475.2/475.2 MB\u001b[0m \u001b[31m2.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m806.7/806.7 kB\u001b[0m \u001b[31m51.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m132.8/132.8 kB\u001b[0m \u001b[31m18.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m17.6/17.6 MB\u001b[0m \u001b[31m79.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m23.7/23.7 MB\u001b[0m \u001b[31m45.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m823.6/823.6 kB\u001b[0m \u001b[31m55.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m14.1/14.1 MB\u001b[0m \u001b[31m68.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m731.7/731.7 MB\u001b[0m \u001b[31m2.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m410.6/410.6 MB\u001b[0m \u001b[31m3.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m121.6/121.6 MB\u001b[0m \u001b[31m8.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m56.5/56.5 MB\u001b[0m \u001b[31m10.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m124.2/124.2 MB\u001b[0m \u001b[31m6.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m196.0/196.0 MB\u001b[0m \u001b[31m6.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m166.0/166.0 MB\u001b[0m \u001b[31m7.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m99.1/99.1 kB\u001b[0m \u001b[31m13.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m167.9/167.9 MB\u001b[0m \u001b[31m6.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m20.5/20.5 MB\u001b[0m \u001b[31m86.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m115.3/115.3 kB\u001b[0m \u001b[31m16.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m134.8/134.8 kB\u001b[0m \u001b[31m19.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m1.6/1.6 MB\u001b[0m \u001b[31m88.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m238.5/238.5 kB\u001b[0m \u001b[31m31.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m55.4/55.4 kB\u001b[0m \u001b[31m7.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m49.4/49.4 kB\u001b[0m \u001b[31m7.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m134.8/134.8 kB\u001b[0m \u001b[31m17.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25h\u001b[31mERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.\n",
            "torchaudio 2.1.0+cu121 requires torch==2.1.0, but you have torch 2.2.0 which is incompatible.\n",
            "torchdata 0.7.0 requires torch==2.1.0, but you have torch 2.2.0 which is incompatible.\n",
            "torchtext 0.16.0 requires torch==2.1.0, but you have torch 2.2.0 which is incompatible.\n",
            "torchvision 0.16.0+cu121 requires torch==2.1.0, but you have torch 2.2.0 which is incompatible.\u001b[0m\u001b[31m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m244.2/244.2 kB\u001b[0m \u001b[31m4.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m72.9/72.9 kB\u001b[0m \u001b[31m5.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m92.5/92.5 MB\u001b[0m \u001b[31m9.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m77.4/77.4 kB\u001b[0m \u001b[31m11.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
            "\u001b[?25h"
          ]
        }
      ],
      "source": [
        "!pip install -q  -U torch datasets transformers tensorflow langchain sentence_transformers faiss-cpu\n",
        "!pip install -q  accelerate==0.21.0 peft==0.4.0 bitsandbytes==0.40.2 trl==0.4.7"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 27,
      "metadata": {
        "id": "q39NlvyOPNni"
      },
      "outputs": [],
      "source": [
        "import os\n",
        "import torch\n",
        "\n",
        "from transformers import (\n",
        "    AutoModelForCausalLM,\n",
        "    AutoTokenizer,\n",
        "    BitsAndBytesConfig,\n",
        "    pipeline\n",
        ")\n",
        "\n",
        "from datasets import load_dataset\n",
        "from peft import LoraConfig, PeftModel\n",
        "\n",
        "from langchain.text_splitter import CharacterTextSplitter\n",
        "from langchain.document_transformers import Html2TextTransformer\n",
        "from langchain.document_loaders import AsyncChromiumLoader, TextLoader\n",
        "from langchain.schema.document import Document\n",
        "\n",
        "from langchain.embeddings.huggingface import HuggingFaceEmbeddings\n",
        "from langchain.vectorstores import FAISS\n",
        "\n",
        "from langchain.prompts import PromptTemplate\n",
        "from langchain.schema.runnable import RunnablePassthrough\n",
        "from langchain.llms import HuggingFacePipeline\n",
        "from langchain.chains import LLMChain\n",
        "\n",
        "import warnings\n",
        "warnings.filterwarnings(\"ignore\")"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "YjH7iwWYg2Rp"
      },
      "source": [
        "## **Model Initialization**"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 3,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 145,
          "referenced_widgets": [
            "af4adf76cdf4419880c741ec94483c7b",
            "f3be6f78f2aa45edbffb8b821b70da88",
            "492f2b6fee904b25b5b3e28bb868c048",
            "2a19d1089dec4136b21f6efbf9ccef8a",
            "e0c241a102344e5fbca4efa4eaf0ebf7",
            "d38ee4334a474d45a7e63d110ef2f768",
            "ecb52d2504f14735b99fdc3aa858aadb",
            "36810915b862435299ef5c6425eba7d1",
            "a4e484f98d3242969a506cc89f519efa",
            "a90cc7bc803c4f9c8a835ef301d411ef",
            "f24322f9d9b74874b463da99d3a8f611",
            "fcdadce3ad594bc99be0118401f3ca8c",
            "05a4e2c3a1014351be98e394d9323cd6",
            "cfa3740a580442f887d3e9c9028b9fd1",
            "aeedc5b8b87c4c78b160624c42a5e2cc",
            "7f49c57c047b42a29754a46efb82953b",
            "3004ec3921304ab7a1cabc942f5c63cd",
            "6a3a252c36b8480880b82d6a1a82aac7",
            "d21ae7381fd148caba9cffebaf57bb50",
            "d9bee2776e2b4d579a422d52985f0fc9",
            "54fb446901f14adab581cbe7352f88f2",
            "1aecfbec46fc4ee19f764c6913b12637",
            "d6ce92fa8be7429bb4677a0d7c398e39",
            "15bf5ca842694c47aeeec35f18e9c281",
            "2d7717ccdca44fca9963d92e6bd23bc0",
            "448214db35414c6ca8af535e7adfb769",
            "b1a9e7d315354d93a8aa745e1eef0f70",
            "0ca2fc4aac03495c8ad0b0b465a1ec72",
            "9df7c050ab2341aaa3b75d80d3024d7f",
            "cc067a51e58743ddbfb318c7d4ad2dcb",
            "181d578663ba4ccf9ed3e628b56b855c",
            "57882ede91ca40549d9fb852c47f0b52",
            "0346077d1ad2492f9a1637666638b8c5",
            "c20d8ac9c3d248b8992922043968c4c9",
            "fc66941a88b64e1a91dce1e1ac2a46e9",
            "704217c1d61a401db13f8ce601343652",
            "0f5f9a8d88fa40d9938777ce5edf99b6",
            "55245f68c6fc41b7ab46232028e76b11",
            "667b04eb852d44a7b1b10c401079d1fd",
            "0e5aff4822b74cb89496ec679a94b53e",
            "fbd610be20c04251a0dbf485906f1a87",
            "b087d738161b484dbbd46f036796d491",
            "ba0c18a32165483cbe74d79b8c609b1e",
            "a062777df055498a8cab521f6d017784"
          ]
        },
        "id": "42fOzXRDPNk1",
        "outputId": "e5f28e96-6b33-41c5-dca4-47ad905b9773"
      },
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "tokenizer_config.json:   0%|          | 0.00/1.47k [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "af4adf76cdf4419880c741ec94483c7b"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "tokenizer.model:   0%|          | 0.00/493k [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "fcdadce3ad594bc99be0118401f3ca8c"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "tokenizer.json:   0%|          | 0.00/1.80M [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "d6ce92fa8be7429bb4677a0d7c398e39"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "special_tokens_map.json:   0%|          | 0.00/72.0 [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "c20d8ac9c3d248b8992922043968c4c9"
            }
          },
          "metadata": {}
        }
      ],
      "source": [
        "#################################################################\n",
        "# Tokenizer and bitsandbytes parameters\n",
        "#################################################################\n",
        "\n",
        "model_name='mistralai/Mistral-7B-Instruct-v0.1'\n",
        "\n",
        "tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)\n",
        "tokenizer.pad_token = tokenizer.eos_token\n",
        "tokenizer.padding_side = \"right\"\n",
        "\n",
        "use_4bit = True\n",
        "bnb_4bit_compute_dtype = \"float16\"\n",
        "bnb_4bit_quant_type = \"nf4\"\n",
        "use_nested_quant = False"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 4,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 241,
          "referenced_widgets": [
            "25b55daba0fc44278df2da684ea50efe",
            "f6b6e77318cc46e193482ca80b2cff14",
            "9e0afa6274494466baf0cec0c733d3f7",
            "906d975b25d74db8b5fc7ddaee6cec39",
            "2eaa234abbec4037b1f0585af2e199e4",
            "bd5c3ca800f04abca2dd2f0c13c90732",
            "8cddfe2e5e574544a478203a7d3f82cf",
            "740c8ea28f1f4d968c7b85182b64d7d6",
            "58ffe8d973d94409a8f2583d20119c5b",
            "c4bbe15133b24b8ea64149dfa0cd2677",
            "33f24463d1d44716bd7b2deff9b9982b",
            "d754e7b034c84b80bfc648819ba9f211",
            "ebe2bd7737b94bb2bc5765b1ac6df47e",
            "f9a298d47faa4d2fa77d80ae18975913",
            "6c83b57669bc4955a5dc9fff5e634ea8",
            "7618e702b27640f096d1f90ae3a16c37",
            "24a0d424a3e748f9b2b49765c045591f",
            "7c2916f53e024124a79ccf7eef43f11e",
            "87b118af30bd42e4b86ee92d954b7814",
            "ddf50a86c002498f9dd33cce7492265b",
            "35a6babf025a4159847bcd0fd50cc2f3",
            "34d96018ca5d4df191286e0a534dbb8a",
            "340937ce8e5b471990c4e2106de493cc",
            "6d1a3070727c434f965d671a7a26ab62",
            "4cadb2a4cf214bc3949a944806838793",
            "92c63f6964e04408ad64867cc487a15a",
            "5cd415c295dc49419517660fa66a8766",
            "2bf79ef2222442c988ac7859f3a5fda1",
            "6f27cddba4f4402e832805406ddacebd",
            "bf5c218414664f0587cf9422854e6d22",
            "745a85fc1244456e9cfd3bf55fd8f572",
            "a384d91a1ae244299489abe77a59e736",
            "a1b21c29cdfc4e4abe823d4cbc999674",
            "752e12e6aff74411bd3d4d2ede8f823f",
            "f73457158a154b668849e528baba91a4",
            "432827e0203146ec92dc3ce02db294b2",
            "3e5b5796603b44dd8f02c05384a2c8e2",
            "84b343699ac145a99b582cb4f887e19b",
            "c40cce2d6bc24f18b4dd1d84f07bd261",
            "139c412dfd964a83b26f482bcfe2fbee",
            "450670fa5c6f407c9111165cf5185255",
            "9a360d8e8bf54ac196d010abdf650ca7",
            "da76cf47245d4b93a2ad9b885b9eea88",
            "1e955068236f4ba081c0eb9c6ef71b97",
            "3cd11826135049ad953abe22d271f699",
            "73b3a8b7a6854fd9a12cb2b0c61e2547",
            "7c8151b8535f47bdb4984d9f5703befb",
            "2953a51914594978a38ebdd5dec512df",
            "b9a25088c4664d3ca0b2c698ae607fc1",
            "502563c04f2b40e498dfe3275cc7ae17",
            "5b4d39ad49d348118bf617fd810fc7cc",
            "90e8389814c54e44bbc3f318fc44268e",
            "690e58bff44e4f56b23b08d2c1c29f4f",
            "7d0fee3a7ab84ffc8b5d12372b17b4bd",
            "7ba446b49de94c2c9ca4dd719d24e4f6",
            "89fc37ff5bc14ff788bfa52e0feee50b",
            "cfe9466f34964556907f2f25ed2baf85",
            "acaad782b1864010a15e61e0f3b5f88b",
            "b9e3a96da0cd42d58ede21665f78fa96",
            "bff09711a14545d3a90a2f6df24eaaac",
            "cd5711916f554825b08f5c7a6a5c2898",
            "825750d43e1244a6bf91b58fb387315c",
            "c48de966b386410eb8c0af1a99969601",
            "bb5710405a0e4334ac95b4a9f5a3140d",
            "628a796150e147a8941b54ba475f6eac",
            "16fa4494aaf748ca954eab6da509eaa6",
            "3c05c29e3a3f46109f1845e13138ac81",
            "b13eb3c954b74e309072be72835d4567",
            "b88a2b03e32f4ae5aa764bb146441454",
            "4f48e4572c28400c827f8932ef5c9dba",
            "212e495e7b774b529cee9e97b9e5be96",
            "40ec30274d1c43d49468d98737b0db4e",
            "08f26b7def6847f98ed8d640c6625fbc",
            "39f8c47fc8e64ff7a9b1ae4c14ade7f9",
            "42a634dadd4a4285b11c4c05ca54958f",
            "c945a280aaa14eeb92e513f87fb65074",
            "a5573b2694264856ac90ace744050714"
          ]
        },
        "id": "3vGBvEeQPNib",
        "outputId": "cdaae447-4de2-446d-e3f8-9d3e7d35ae45"
      },
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "config.json:   0%|          | 0.00/571 [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "25b55daba0fc44278df2da684ea50efe"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "model.safetensors.index.json:   0%|          | 0.00/25.1k [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "d754e7b034c84b80bfc648819ba9f211"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Downloading shards:   0%|          | 0/2 [00:00<?, ?it/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "340937ce8e5b471990c4e2106de493cc"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "model-00001-of-00002.safetensors:   0%|          | 0.00/9.94G [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "752e12e6aff74411bd3d4d2ede8f823f"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "model-00002-of-00002.safetensors:   0%|          | 0.00/4.54G [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "3cd11826135049ad953abe22d271f699"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Loading checkpoint shards:   0%|          | 0/2 [00:00<?, ?it/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "89fc37ff5bc14ff788bfa52e0feee50b"
            }
          },
          "metadata": {}
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "generation_config.json:   0%|          | 0.00/116 [00:00<?, ?B/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "3c05c29e3a3f46109f1845e13138ac81"
            }
          },
          "metadata": {}
        }
      ],
      "source": [
        "#################################################################\n",
        "# Set up quantization config\n",
        "#################################################################\n",
        "compute_dtype = getattr(torch, bnb_4bit_compute_dtype)\n",
        "\n",
        "bnb_config = BitsAndBytesConfig(\n",
        "    load_in_4bit=use_4bit,\n",
        "    bnb_4bit_quant_type=bnb_4bit_quant_type,\n",
        "    bnb_4bit_compute_dtype=compute_dtype,\n",
        "    bnb_4bit_use_double_quant=use_nested_quant,\n",
        ")\n",
        "\n",
        "# Check GPU compatibility with bfloat16\n",
        "if compute_dtype == torch.float16 and use_4bit:\n",
        "    major, _ = torch.cuda.get_device_capability()\n",
        "    if major >= 8:\n",
        "        print(\"=\" * 80)\n",
        "        print(\"Your GPU supports bfloat16: accelerate training with bf16=True\")\n",
        "        print(\"=\" * 80)\n",
        "\n",
        "#################################################################\n",
        "# Load pre-trained config\n",
        "#################################################################\n",
        "model = AutoModelForCausalLM.from_pretrained(\n",
        "    model_name,\n",
        "    quantization_config=bnb_config,\n",
        ")"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 5,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "i4L5NroyPNfp",
        "outputId": "a57f41a5-b803-43ce-99c8-5ad79f8ed242"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "trainable model parameters: 262410240\n",
            "all model parameters: 3752071168\n",
            "percentage of trainable model parameters: 6.99%\n"
          ]
        }
      ],
      "source": [
        "def print_number_of_trainable_model_parameters(model):\n",
        "    trainable_model_params = 0\n",
        "    all_model_params = 0\n",
        "    for _, param in model.named_parameters():\n",
        "        all_model_params += param.numel()\n",
        "        if param.requires_grad:\n",
        "            trainable_model_params += param.numel()\n",
        "    return f\"trainable model parameters: {trainable_model_params}\\nall model parameters: {all_model_params}\\npercentage of trainable model parameters: {100 * trainable_model_params / all_model_params:.2f}%\"\n",
        "\n",
        "print(print_number_of_trainable_model_parameters(model))"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 6,
      "metadata": {
        "id": "ANsVO0vjPiCL"
      },
      "outputs": [],
      "source": [
        "text_generation_pipeline = pipeline(\n",
        "    model=model,\n",
        "    tokenizer=tokenizer,\n",
        "    task=\"text-generation\",\n",
        "    temperature=0.1,\n",
        "    repetition_penalty=1.0,\n",
        "    return_full_text=True,\n",
        "    max_new_tokens=2048,\n",
        ")"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 7,
      "metadata": {
        "id": "KmCjpYRkPh9j"
      },
      "outputs": [],
      "source": [
        "mistral_llm = HuggingFacePipeline(pipeline=text_generation_pipeline)"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "1LTISSIIg7_n"
      },
      "source": [
        "## **RAG Setup**"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "JKV1hhabPh7K"
      },
      "outputs": [],
      "source": [
        "# !playwright install\n",
        "# !playwright install-deps"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "iDbRpgLmPNc2"
      },
      "outputs": [],
      "source": [
        "# import nest_asyncio\n",
        "# nest_asyncio.apply()\n",
        "\n",
        "# # Articles to index\n",
        "# articles = [\n",
        "#   'https://www.w3schools.com/css/css_grid.asp',\n",
        "#   'https://www.w3schools.com/css/css_grid_container.asp',\n",
        "#   'https://www.w3schools.com/css/css_grid_item.asp',\n",
        "#   'https://tailwindcss.com/docs/installation',\n",
        "# ]\n",
        "\n",
        "# loader = AsyncChromiumLoader(articles)\n",
        "# docs = loader.load()\n",
        "\n",
        "# # Converts HTML to plain text\n",
        "# html2text = Html2TextTransformer()\n",
        "# docs_transformed = html2text.transform_documents(docs)\n",
        "\n",
        "# # Chunk text\n",
        "# text_splitter = CharacterTextSplitter(chunk_size=1024,\n",
        "#                                       chunk_overlap=0)\n",
        "# chunked_documents = text_splitter.split_documents(docs_transformed)\n",
        "\n",
        "# # Load chunked documents into the FAISS index\n",
        "# db = FAISS.from_documents(chunked_documents,\n",
        "#                           HuggingFaceEmbeddings(model_name='sentence-transformers/all-mpnet-base-v2'))\n",
        "\n",
        "# retriever = db.as_retriever()"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "embeddings = HuggingFaceEmbeddings()\n",
        "db = FAISS.load_local(folder_path='./bootstrap_rag', embeddings=embeddings)\n",
        "retriever = db.as_retriever()"
      ],
      "metadata": {
        "id": "D9OIQL-qh4TG"
      },
      "execution_count": 9,
      "outputs": []
    },
    {
      "cell_type": "code",
      "execution_count": 10,
      "metadata": {
        "id": "KgYs9T1sT6G6",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "289f992e-a5ce-4066-ae75-76f3a5388bad"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "tags=['FAISS', 'HuggingFaceEmbeddings'] vectorstore=<langchain_community.vectorstores.faiss.FAISS object at 0x7e0778263c70>\n"
          ]
        }
      ],
      "source": [
        "print(retriever)"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "# db.save_local(\"bootstrap_embeddings\")"
      ],
      "metadata": {
        "id": "xNHkhfoPHB4q"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "ea3d18GVhB4a"
      },
      "source": [
        "## **Prompt Templating**"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 11,
      "metadata": {
        "id": "uMe52_bNP0FM"
      },
      "outputs": [],
      "source": [
        "prompt_template = \"\"\"\n",
        "### [INST] Instruction: You are a UI generating assistant. You'll carefully analyse the input prompt and then create the UI using React and CSS3. Use CSS classes wherever you can and define them in a styles tag.\n",
        "Infer things like background colors, shadows, borders, etc from the nature of the UI component. Follow the input prompt thoroughly. Print only the code that is asked in the input prompt\n",
        "and nothing else (no explanation or comments or things like \"here you go\", \"here's the code that you asked for\", etc)\n",
        "\n",
        "{context}\n",
        "\n",
        "### QUESTION:\n",
        "{question} [/INST]\n",
        " \"\"\"\n",
        "\n",
        "# Create prompt from prompt template\n",
        "prompt = PromptTemplate(\n",
        "    input_variables=[\"context\", \"question\"],\n",
        "    template=prompt_template,\n",
        ")\n",
        "\n",
        "# Create llm chain\n",
        "llm_chain = LLMChain(llm=mistral_llm, prompt=prompt)\n",
        "\n",
        "rag_chain = (\n",
        " {\"context\": retriever, \"question\": RunnablePassthrough()}\n",
        "    | llm_chain\n",
        ")"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "GO8DNFBchFiZ"
      },
      "source": [
        "### **Test Output**"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "mHjUHMMgUx5z"
      },
      "source": [
        "##### **Without RAG**"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "Hhjz0v1LP0Cr",
        "outputId": "cca3a24e-d211-449d-a680-62c35fa074ee"
      },
      "outputs": [
        {
          "name": "stderr",
          "output_type": "stream",
          "text": [
            "Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.\n"
          ]
        },
        {
          "name": "stdout",
          "output_type": "stream",
          "text": [
            "```jsximport React from'react';import './styles.css';const Card = ({ heading, text }) => {  return (    <div className=\"card\">      <h2 className=\"heading\">{heading}</h2>      <p className=\"text\">{text}</p>    </div>  );};export default Card;```\n"
          ]
        }
      ],
      "source": [
        "question = '''\n",
        "Generate a simple card with a heading and some sample text (use lorem ipsum).\n",
        "Give the react component as output and follow the styles given as context.\n",
        "'''\n",
        "\n",
        "context = ''\n",
        "\n",
        "result = llm_chain.invoke({\"context\": context, \"question\": question})\n",
        "\n",
        "print(result['text'].replace('\\n', ''))"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "y4t7JCDhU0Mm"
      },
      "source": [
        "##### **With RAG**"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "2B8JFJf0P0AV"
      },
      "outputs": [],
      "source": [
        "rag_chain = (\n",
        " {\"context\": retriever, \"question\": RunnablePassthrough()}\n",
        "    | llm_chain\n",
        ")\n",
        "\n",
        "question = '''\n",
        "Generate a simple card with a heading and some sample text. Infer from the context the heading and a sample text.\n",
        "Give the react component as output and follow the styles given as context.\n",
        "'''\n",
        "\n",
        "result = rag_chain.invoke(question)\n",
        "\n",
        "print(result['text'])"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "ZLd3rqb8hQe2"
      },
      "source": [
        "## **UI Generation**"
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "### **Simple Components**"
      ],
      "metadata": {
        "id": "txJbRuiqv9Zj"
      }
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "GOeW26-QWl9u"
      },
      "source": [
        "##### **Card Component**"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "BoJdrujwPz7C",
        "outputId": "8a9ed9cd-ae9b-4a52-dee8-e206c7ed539d"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\n",
            "```javascript\n",
            "import React from'react';\n",
            "import { addComponents } from 'tailwindcss-in-js';\n",
            "\n",
            "const Card = ({ img, imgAlt }) => {\n",
            "  const classes = addComponents({\n",
            "    '.card': {\n",
            "      backgroundColor: '#fff',\n",
            "      borderRadius: '.25rem',\n",
            "      boxShadow: '0 2px 4px rgba(0,0,0,0.2)',\n",
            "      '&:hover': {\n",
            "        boxShadow: '0 10px 15px rgba(0,0,0,0.2)',\n",
            "      },\n",
            "      '@media (min-width: 500px)': {\n",
            "        borderRadius: '.5rem',\n",
            "      },\n",
            "    },\n",
            "    '.card-image': {\n",
            "      width: '100%',\n",
            "      height: 'auto',\n",
            "      objectFit: 'cover',\n",
            "    },\n",
            "    '.card-text': {\n",
            "      textAlign: 'center',\n",
            "      fontSize: '1.2rem',\n",
            "      fontWeight: 'bold',\n",
            "      color: '#000',\n",
            "      marginTop: '1rem',\n",
            "    },\n",
            "  });\n",
            "\n",
            "  return (\n",
            "    <div className={classes.card}>\n",
            "      <img className={classes.cardImage} src={img} alt={imgAlt} />\n",
            "      <div className={classes.cardText}>Sample Image</div>\n",
            "    </div>\n",
            "  );\n",
            "};\n",
            "\n",
            "export default Card;\n",
            "```\n"
          ]
        }
      ],
      "source": [
        "question = '''\n",
        "Generate a simple image card. The top of the card should have an image with source as : https://a-z-animals.com/media/2022/11/shutterstock_606517310-1024x650.jpg.\n",
        "The bottom of the card should contain the text \"Sample Image\". Give the react component as output and follow the styles given in the context.\n",
        "'''\n",
        "\n",
        "result = rag_chain.invoke(question)\n",
        "\n",
        "print(result['text'])"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 12,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "omnUf_7y9idj",
        "outputId": "72b83b11-ed6b-4af2-c1f8-6d1ce7801781"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\n",
            "```\n",
            "import React from'react';\n",
            "import './ImageCard.css';\n",
            "\n",
            "const ImageCard = () => {\n",
            "  return (\n",
            "    <div className=\"image-card\">\n",
            "      <img src=\"https://shorturl.at/kCOZ2\" alt=\"Sample Image\" className=\"image\" />\n",
            "      <p className=\"text\">Sample Image</p>\n",
            "    </div>\n",
            "  );\n",
            "};\n",
            "\n",
            "export default ImageCard;\n",
            "```\n",
            "\n",
            "```\n",
            ".image-card {\n",
            "  display: flex;\n",
            "  flex-direction: column;\n",
            "  align-items: center;\n",
            "  justify-content: center;\n",
            "  width: 300px;\n",
            "  height: 400px;\n",
            "  background-color: #f2f2f2;\n",
            "  border-radius: 10px;\n",
            "  box-shadow: 0px 0px 10px rgba(0, 0, 0, 0.2);\n",
            "}\n",
            "\n",
            ".image {\n",
            "  width: 100%;\n",
            "  height: 200px;\n",
            "  object-fit: cover;\n",
            "  border-radius: 10px;\n",
            "}\n",
            "\n",
            ".text {\n",
            "  font-size: 18px;\n",
            "  text-align: center;\n",
            "  color: #333;\n",
            "  margin-top: 10px;\n",
            "}\n",
            "```\n"
          ]
        }
      ],
      "source": [
        "rag_chain = (\n",
        " {\"context\": retriever, \"question\": RunnablePassthrough()}\n",
        "    | llm_chain\n",
        ")\n",
        "\n",
        "question = '''\n",
        "Generate a simple image card. The top of the card should have an image with source as : https://shorturl.at/kCOZ2.\n",
        "The bottom of the card should contain the text \"Sample Image\".\n",
        "Give the react component as output and follow the styles given in the context.\n",
        "DO NOT USE ANY PROPS instead hardcode the values wherever necessary.\n",
        "'''\n",
        "\n",
        "result = rag_chain.invoke(question)\n",
        "\n",
        "print(result['text'])"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 14,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "nDvQ6B459ibE",
        "outputId": "227266fc-3085-444b-91dd-cde95366aec5"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\n",
            "```\n",
            "import React from'react';\n",
            "import './Card.css';\n",
            "\n",
            "const Card = () => {\n",
            "  return (\n",
            "    <div className=\"card\">\n",
            "      <img src=\"https://shorturl.at/kCOZ2\" alt=\"Sample Image\" className=\"card-image\" />\n",
            "      <div className=\"card-text\">\n",
            "        <h2 className=\"card-heading\">Sample Image</h2>\n",
            "        <p className=\"card-body\">\n",
            "          In publishing and graphic design, Lorem ipsum is a placeholder text commonly used to demonstrate the visual form of a document or a typeface without relying on meaningful content.\n",
            "        </p>\n",
            "        <button className=\"card-button\">Read More</button>\n",
            "      </div>\n",
            "    </div>\n",
            "  );\n",
            "};\n",
            "\n",
            "export default Card;\n",
            "```\n",
            "\n",
            "```\n",
            ".card {\n",
            "  display: flex;\n",
            "  flex-direction: column;\n",
            "  align-items: center;\n",
            "  background-color: #f2f2f2;\n",
            "  border-radius: 5px;\n",
            "  box-shadow: 0 0 5px rgba(0, 0, 0, 0.2);\n",
            "  margin: 10px;\n",
            "  width: 300px;\n",
            "  height: 400px;\n",
            "}\n",
            "\n",
            ".card-image {\n",
            "  width: 100%;\n",
            "  height: 200px;\n",
            "  object-fit: cover;\n",
            "}\n",
            "\n",
            ".card-text {\n",
            "  display: flex;\n",
            "  flex-direction: column;\n",
            "  align-items: center;\n",
            "  justify-content: space-between;\n",
            "  padding: 10px;\n",
            "  width: 100%;\n",
            "  height: 200px;\n",
            "}\n",
            "\n",
            ".card-heading {\n",
            "  font-size: 20px;\n",
            "  font-weight: bold;\n",
            "  margin-bottom: 10px;\n",
            "}\n",
            "\n",
            ".card-body {\n",
            "  font-size: 15px;\n",
            "  line-height: 1.5;\n",
            "  margin-bottom: 10px;\n",
            "}\n",
            "\n",
            ".card-button {\n",
            "  background-color: #4CAF50;\n",
            "  color: white;\n",
            "  font-size: 16px;\n",
            "  padding: 10px 20px;\n",
            "  text-align: center;\n",
            "  text-decoration: none;\n",
            "  border: none;\n",
            "  border-radius: 5px;\n",
            "  cursor: pointer;\n",
            "  margin-top: 10px;\n",
            "}\n",
            "\n",
            ".card-button:hover {\n",
            "  background-color: #3e8e41;\n",
            "}\n",
            "```\n"
          ]
        }
      ],
      "source": [
        "rag_chain = (\n",
        " {\"context\": retriever, \"question\": RunnablePassthrough()}\n",
        "    | llm_chain\n",
        ")\n",
        "\n",
        "question = '''\n",
        "Generate a simple card. The top of the card should have an image with source as : https://shorturl.at/kCOZ2.\n",
        "The bottom of the card should contain the text \"Sample Image\". This is the heading (at most 20px).\n",
        "Below that, there should be a smaller text (1.5 times smaller) that says \"In publishing and graphic design, Lorem ipsum\n",
        "is a placeholder text commonly used to demonstrate the visual form of a document or a typeface without relying on meaningful content.\" Below that, there should be\n",
        "a simple button that says 'Read More'.\n",
        "\n",
        "Give the react component as output and follow the styles given in the context. DO NOT USE ANY PROPS\n",
        "instead hardcode the values wherever necessary.\n",
        "'''\n",
        "\n",
        "result = rag_chain.invoke(question)\n",
        "\n",
        "print(result['text'])"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "qeHarxVH9iYe"
      },
      "outputs": [],
      "source": [
        "rag_chain = (\n",
        " {\"context\": retriever, \"question\": RunnablePassthrough()}\n",
        "    | llm_chain\n",
        ")\n",
        "\n",
        "question = '''\n",
        "Generate a simple product card. The top of the card should have an image with source as : https://shorturl.at/kmzN2. For the bottom part, use a 2-column layout.\n",
        "The bottom of card should contain the following : \"Sample Shoes\" as heading, \"Sample Company\" as subheading in a lighter text and \"$120.00\" as price\n",
        "in a bold font. All of them should be below each other and should be in the first column. In the second column, right beside the heading, include a button that says \"+\".\n",
        "\n",
        "Give the react component as output and follow the styles given in the context. DO NOT USE ANY PROPS instead hardcode the values wherever necessary.\n",
        "'''\n",
        "\n",
        "result = rag_chain.invoke(question)\n",
        "\n",
        "print(result['text'])"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "PM0pkNgIWuEm"
      },
      "source": [
        "##### **Input Component**"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 15,
      "metadata": {
        "id": "iyoUB4CP9iWL",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "c2fbeec2-08ec-4d3b-b72d-115342be91dc"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\n",
            "```jsx\n",
            "import React from'react';\n",
            "import './ContactForm.css';\n",
            "\n",
            "const ContactForm = () => {\n",
            "  return (\n",
            "    <div className=\"contact-form-container\">\n",
            "      <div className=\"contact-form-title\">Contact Form</div>\n",
            "      <div className=\"contact-form-row\">\n",
            "        <div className=\"contact-form-column\">\n",
            "          <input\n",
            "            type=\"text\"\n",
            "            placeholder=\"Sample\"\n",
            "            className=\"contact-form-input\"\n",
            "          />\n",
            "          <label className=\"contact-form-label\">Name</label>\n",
            "        </div>\n",
            "      </div>\n",
            "      <div className=\"contact-form-row\">\n",
            "        <div className=\"contact-form-column\">\n",
            "          <input\n",
            "            type=\"email\"\n",
            "            placeholder=\"sample@gmail.com\"\n",
            "            className=\"contact-form-input\"\n",
            "          />\n",
            "          <label className=\"contact-form-label\">Email</label>\n",
            "        </div>\n",
            "      </div>\n",
            "      <div className=\"contact-form-row\">\n",
            "        <div className=\"contact-form-column\">\n",
            "          <button className=\"contact-form-button\">Submit</button>\n",
            "        </div>\n",
            "      </div>\n",
            "    </div>\n",
            "  );\n",
            "};\n",
            "\n",
            "export default ContactForm;\n",
            "```\n",
            "\n",
            "```css\n",
            ".contact-form-container {\n",
            "  max-height: 800px;\n",
            "  max-width: 500px;\n",
            "  display: flex;\n",
            "  flex-direction: column;\n",
            "  align-items: center;\n",
            "  justify-content: center;\n",
            "  background-color: #f2f2f2;\n",
            "  border-radius: 10px;\n",
            "  box-shadow: 0px 0px 10px rgba(0, 0, 0, 0.2);\n",
            "}\n",
            "\n",
            ".contact-form-title {\n",
            "  font-size: 24px;\n",
            "  font-weight: bold;\n",
            "  text-align: center;\n",
            "  margin-bottom: 20px;\n",
            "}\n",
            "\n",
            ".contact-form-row {\n",
            "  display: flex;\n",
            "  flex-direction: row;\n",
            "  align-items: center;\n",
            "  justify-content: center;\n",
            "  width: 100%;\n",
            "}\n",
            "\n",
            ".contact-form-column {\n",
            "  flex: 1;\n",
            "  margin: 10px;\n",
            "}\n",
            "\n",
            ".contact-form-input {\n",
            "  padding: 10px;\n",
            "  font-size: 16px;\n",
            "  border: none;\n",
            "  border-radius: 5px;\n",
            "  box-shadow: 0px 0px 5px rgba(0, 0, 0, 0.2);\n",
            "}\n",
            "\n",
            ".contact-form-label {\n",
            "  font-size: 16px;\n",
            "  font-weight: bold;\n",
            "  margin-bottom: 5px;\n",
            "}\n",
            "\n",
            ".contact-form-button {\n",
            "  padding: 10px 20px;\n",
            "  font-size: 16px;\n",
            "  border: none;\n",
            "  border-radius: 5px;\n",
            "  background-color: #4CAF50;\n",
            "  color: white;\n",
            "  cursor: pointer;\n",
            "}\n",
            "\n",
            ".contact-form-button:hover {\n",
            "  background-color: #3e8e41;\n",
            "}\n",
            "```\n"
          ]
        }
      ],
      "source": [
        "question = '''\n",
        "Generate a simple form component. Give the form a title (heading) of \"Contact Form\". This should follow a 12 column layout. Inside this form component have the following in separate rows :\n",
        "- input component with placeholder as \"Sample\" and label as \"Name\". It should spans 12 columns\n",
        "- input component with placeholder as \"sample@gmail.com\" and label as \"Email\". It should spans 12 columns\n",
        "- one button component with text \"Submit\" at the bottom of the container.\n",
        "\n",
        "Make the form responsive and make sure maximum dimensions are height = 800px and width = 500px.\n",
        "\n",
        "Give the react component (JS not typescript) as output and follow the styles given in the context.\n",
        "DO NOT USE ANY PROPS instead hardcode the values wherever necessary.\n",
        "Define all the functions that you use in the component definition.\n",
        "DO NOT USE ANY OTHER ADDITIONAL LIBRARIES for icons, etc.\n",
        "\n",
        "'''\n",
        "\n",
        "result = rag_chain.invoke(question)\n",
        "\n",
        "print(result['text'])"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 17,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "HGu91tHR9iTh",
        "outputId": "9ce09597-4b84-4c22-e7b8-171f97c89ff8"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\n",
            "```jsx\n",
            "import React from'react';\n",
            "import './styles.css';\n",
            "\n",
            "const ContactForm = () => {\n",
            "  return (\n",
            "    <div className=\"container\">\n",
            "      <div className=\"row\">\n",
            "        <div className=\"col-md-12\">\n",
            "          <h1 className=\"text-center mb-5\">Contact Form</h1>\n",
            "          <form className=\"form-group\">\n",
            "            <div className=\"row\">\n",
            "              <div className=\"col-md-6\">\n",
            "                <label className=\"form-label\">Name</label>\n",
            "                <input type=\"text\" className=\"form-control\" placeholder=\"Sample\" />\n",
            "              </div>\n",
            "              <div className=\"col-md-6\">\n",
            "                <label className=\"form-label\">Email</label>\n",
            "                <input type=\"email\" className=\"form-control\" placeholder=\"sample@gmail.com\" />\n",
            "              </div>\n",
            "            </div>\n",
            "            <div className=\"row\">\n",
            "              <div className=\"col-md-8\">\n",
            "                <label className=\"form-label\">Address</label>\n",
            "                <input type=\"text\" className=\"form-control\" placeholder=\"Sample Address\" />\n",
            "              </div>\n",
            "              <div className=\"col-md-4\">\n",
            "                <label className=\"form-label\">Pincode</label>\n",
            "                <input type=\"text\" className=\"form-control\" placeholder=\"Sample PIN\" />\n",
            "              </div>\n",
            "            </div>\n",
            "            <div className=\"row\">\n",
            "              <div className=\"col-md-12\">\n",
            "                <button type=\"submit\" className=\"btn btn-primary btn-block\">Submit</button>\n",
            "              </div>\n",
            "            </div>\n",
            "          </form>\n",
            "        </div>\n",
            "      </div>\n",
            "    </div>\n",
            "  );\n",
            "};\n",
            "\n",
            "export default ContactForm;\n",
            "```\n",
            "\n",
            "```css\n",
            ".container {\n",
            "  max-height: 800px;\n",
            "  max-width: 500px;\n",
            "  padding: 20px;\n",
            "  margin: 0 auto;\n",
            "}\n",
            "\n",
            ".row {\n",
            "  display: flex;\n",
            "  flex-wrap: wrap;\n",
            "}\n",
            "\n",
            ".col-md-12 {\n",
            "  flex-basis: 100%;\n",
            "  padding: 20px;\n",
            "}\n",
            "\n",
            ".form-group {\n",
            "  display: flex;\n",
            "  flex-wrap: wrap;\n",
            "}\n",
            "\n",
            ".col-md-6 {\n",
            "  flex-basis: 50%;\n",
            "  padding: 20px;\n",
            "}\n",
            "\n",
            ".form-label {\n",
            "  font-weight: bold;\n",
            "  margin-bottom: 10px;\n",
            "}\n",
            "\n",
            ".form-control {\n",
            "  padding: 10px;\n",
            "  border-radius: 5px;\n",
            "  border: none;\n",
            "  box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);\n",
            "}\n",
            "\n",
            ".btn-block {\n",
            "  width: 100%;\n",
            "}\n",
            "\n",
            ".btn {\n",
            "  padding: 10px 20px;\n",
            "  border-radius: 5px;\n",
            "  border: none;\n",
            "  background-color: #4CAF50;\n",
            "  color: white;\n",
            "  cursor: pointer;\n",
            "}\n",
            "\n",
            ".btn:hover {\n",
            "  background-color: #3e8e41;\n",
            "}\n",
            "```\n"
          ]
        }
      ],
      "source": [
        "question = '''\n",
        "Generate a simple form component. Give the form a title (heading) of \"Contact Form\". This should follow a 12 column layout and should be left aligned.\n",
        "Inside this form component have the following :\n",
        "first row :\n",
        "- input component with placeholder as \"Sample\" and label as \"Name\". It should spans 6 columns\n",
        "- input component with placeholder as \"sample@gmail.com\" and label as \"Email\". It should spans 6 columns\n",
        "second row :\n",
        "- input component with placeholder as \"Sample Address\" and label as \"Address\". It should spans 8 columns\n",
        "- input component with placeholder as \"Sample PIN\" and label as \"Pincode\". It should spans 4 columns\n",
        "last row :\n",
        "- one button component with text \"Submit\". It should span 12 columns.\n",
        "\n",
        "Make the form responsive and make sure maximum dimensions are height = 800px and width = 500px with decent padding.\n",
        "Give the react component (JS not typescript) as output and follow the styles given in the context.\n",
        "Define all the functions that you use in the component definition.\n",
        "DO NOT USE ANY PROPS instead hardcode the values wherever necessary.\n",
        "\n",
        "ALLOWED : import React from'react';\n",
        "ALLOWED : import \"styles.css\";\n",
        "\n",
        "NO OTHER IMPORTS ARE ALLOWED.\n",
        "FOLLOW THE COLUMN SPANS USING BOOTSTRAP FLEX-BOX OR GRID.\n",
        "GIVE PROPER PADDINGS AND MARGIN WHEREVER NECESSARY SO THAT THE UI LOOKS GOOD.\n",
        "\n",
        "'''\n",
        "\n",
        "result = rag_chain.invoke(question)\n",
        "\n",
        "print(result['text'])"
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "### **Complex Components**"
      ],
      "metadata": {
        "id": "kbNR-nvpskwF"
      }
    },
    {
      "cell_type": "markdown",
      "source": [
        "##### **Pricing Plan**"
      ],
      "metadata": {
        "id": "NPV-67fNv1u2"
      }
    },
    {
      "cell_type": "code",
      "execution_count": 18,
      "metadata": {
        "id": "18D4d5w-9iOe",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "4e66e032-ebb6-451d-a6c0-797e9bbe08c4"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\n",
            "```jsx\n",
            "import React from'react';\n",
            "import './styles.css';\n",
            "\n",
            "const PricingPage = () => {\n",
            "  return (\n",
            "    <section className=\"pricing-page\">\n",
            "      <h1 className=\"pricing-page-title\">Pricing</h1>\n",
            "      <p className=\"pricing-page-copy\">Lorem ipsum may be used the final copy is available. Lorem ipsum may be used the final copy is available.</p>\n",
            "      <div className=\"pricing-cards\">\n",
            "        <div className=\"pricing-card\">\n",
            "          <h2 className=\"pricing-card-header\">Free</h2>\n",
            "          <p className=\"pricing-card-price\">0$ / month</p>\n",
            "          <ul className=\"pricing-card-list\">\n",
            "            <li>Lorem ipsum may be used the final copy is available.</li>\n",
            "            <li>as a placeholder before</li>\n",
            "            <li>sample text here</li>\n",
            "          </ul>\n",
            "          <button className=\"pricing-card-cta\">Download Now</button>\n",
            "        </div>\n",
            "        <div className=\"pricing-card\">\n",
            "          <h2 className=\"pricing-card-header\">Pro</h2>\n",
            "          <p className=\"pricing-card-price\">$8 / month</p>\n",
            "          <ul className=\"pricing-card-list\">\n",
            "            <li>Lorem ipsum may be used the final copy is available.</li>\n",
            "            <li>as a placeholder before</li>\n",
            "            <li>sample text here</li>\n",
            "          </ul>\n",
            "          <button className=\"pricing-card-cta\">Buy Now</button>\n",
            "        </div>\n",
            "        <div className=\"pricing-card\">\n",
            "          <h2 className=\"pricing-card-header\">Ultra</h2>\n",
            "          <p className=\"pricing-card-price\">$20 / month</p>\n",
            "          <ul className=\"pricing-card-list\">\n",
            "            <li>Lorem ipsum may be used the final copy is available.</li>\n",
            "            <li>as a placeholder before</li>\n",
            "            <li>sample text here</li>\n",
            "          </ul>\n",
            "          <button className=\"pricing-card-cta\">Contact Sales</button>\n",
            "        </div>\n",
            "      </div>\n",
            "    </section>\n",
            "  );\n",
            "};\n",
            "\n",
            "export default PricingPage;\n",
            "```\n",
            "\n",
            "```css\n",
            ".pricing-page {\n",
            "  background-color: #f5f5f5;\n",
            "  padding: 2rem;\n",
            "}\n",
            "\n",
            ".pricing-page-title {\n",
            "  font-size: 2rem;\n",
            "  margin-bottom: 1rem;\n",
            "}\n",
            "\n",
            ".pricing-page-copy {\n",
            "  font-size: 1.2rem;\n",
            "  margin-bottom: 2rem;\n",
            "}\n",
            "\n",
            ".pricing-cards {\n",
            "  display: flex;\n",
            "  flex-wrap: wrap;\n",
            "  justify-content: space-between;\n",
            "  margin-bottom: 2rem;\n",
            "}\n",
            "\n",
            ".pricing-card {\n",
            "  background-color: #fff;\n",
            "  box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);\n",
            "  padding: 1.5rem;\n",
            "  width: calc(33.333% - 1rem);\n",
            "}\n",
            "\n",
            ".pricing-card-header {\n",
            "  font-size: 1.5rem;\n",
            "  font-weight: bold;\n",
            "  margin-bottom: 1rem;\n",
            "}\n",
            "\n",
            ".pricing-card-price {\n",
            "  font-size: 1.2rem;\n",
            "  font-weight: bold;\n",
            "  margin-bottom: 1rem;\n",
            "}\n",
            "\n",
            ".pricing-card-list {\n",
            "  list-style: none;\n",
            "  padding: 0;\n",
            "}\n",
            "\n",
            ".pricing-card-list li {\n",
            "  font-size: 1rem;\n",
            "  margin-bottom: 0.5rem;\n",
            "}\n",
            "\n",
            ".pricing-card-cta {\n",
            "  background-color: #007bff;\n",
            "  color: #fff;\n",
            "  padding: 0.5rem 1rem;\n",
            "  text-decoration: none;\n",
            "  border: none;\n",
            "  cursor: pointer;\n",
            "}\n",
            "\n",
            ".pricing-card-cta:hover {\n",
            "  background-color: #0056b3;\n",
            "}\n",
            "```\n"
          ]
        }
      ],
      "source": [
        "question = '''\n",
        "Generate a pricing page component. This should be a section. The title should say \"Pricing\". Below that,\n",
        "in a smaller font-size, there should a copy that says \"Lorem ipsum may be used the final copy is available. Lorem ipsum may be used the final copy is available\".\n",
        "Inside that section, we should have 3 pricing cards.\n",
        "The pricing card contains a header that has the tiers \"Free\", \"Pro\", \"Ultra\". This is an accent text.\n",
        "This should be separated by a line. Below that, there should be the price \"0$ / month\", \"$8 / month\", \"$20 / month\".\n",
        "This should have the heading font size. Below that, include the following text as a list for each of the cards :\n",
        "\"Lorem ipsum may be used the final copy is available.\"\n",
        "\"as a placeholder before\"\n",
        "\"sample text here\"\n",
        "Prettify the list and make sure the colors are pleasing to the ease.\n",
        "Below that there should be a CTA button that says \"Download Now\", \"Buy Now\", \"Contact Sales\" in those 3 pricing cards.\n",
        "Have a light grey as background and make the card white with decent shadows that look good.\n",
        "\n",
        "Give the react component (JS not typescript) as output and follow the styles given in the context.\n",
        "Define all the functions that you use in the component definition.\n",
        "DO NOT USE ANY PROPS instead hardcode the values wherever necessary.\n",
        "\n",
        "ALLOWED : import React from'react';\n",
        "ALLOWED : import \"styles.css\";\n",
        "\n",
        "NO OTHER IMPORTS ARE ALLOWED.\n",
        "FOLLOW THE COLUMN SPANS USING BOOTSTRAP FLEX-BOX OR GRID.\n",
        "GIVE PROPER PADDINGS AND MARGIN WHEREVER NECESSARY SO THAT THE UI LOOKS GOOD.\n",
        "GIVE PROPER FONT-SIZE, FONT-COLOR AND OTHER FONT RELATED THINGS SO THAT THE UI LOOKS GOOD\n",
        "GIVE PROPER COLORS, SHADOWS, ACCENTS, ETC SO THAT THE UI LOOKS GOOD.\n",
        "\n",
        "'''\n",
        "\n",
        "result = rag_chain.invoke(question)\n",
        "\n",
        "print(result['text'])"
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "##### **List Group**"
      ],
      "metadata": {
        "id": "wRK1SXQ3zmqA"
      }
    },
    {
      "cell_type": "code",
      "execution_count": 19,
      "metadata": {
        "id": "sD3JvQTIWryZ",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "49384cfb-8b4a-42fa-b1ea-76eb21dd8fd4"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\n",
            "```jsx\n",
            "import React from'react';\n",
            "import './styles.css';\n",
            "\n",
            "const ListGroup = () => {\n",
            "  return (\n",
            "    <div className=\"container\">\n",
            "      <div className=\"row\">\n",
            "        <div className=\"col-md-4\">\n",
            "          <div className=\"list-group\">\n",
            "            <div className=\"list-group-item\">\n",
            "              <img src=\"https://shorturl.at/jntJT\" className=\"avatar\" />\n",
            "              <div className=\"list-group-item-content\">\n",
            "                <h3 className=\"title\">Lorem ipsum</h3>\n",
            "                <p className=\"description\">Sample description</p>\n",
            "              </div>\n",
            "            </div>\n",
            "            <div className=\"list-group-item\">\n",
            "              <img src=\"https://shorturl.at/jntJT\" className=\"avatar\" />\n",
            "              <div className=\"list-group-item-content\">\n",
            "                <h3 className=\"title\">Lorem ipsum</h3>\n",
            "                <p className=\"description\">Sample description</p>\n",
            "              </div>\n",
            "            </div>\n",
            "            <div className=\"list-group-item\">\n",
            "              <img src=\"https://shorturl.at/jntJT\" className=\"avatar\" />\n",
            "              <div className=\"list-group-item-content\">\n",
            "                <h3 className=\"title\">Lorem ipsum</h3>\n",
            "                <p className=\"description\">Sample description</p>\n",
            "              </div>\n",
            "            </div>\n",
            "            <div className=\"list-group-item\">\n",
            "              <img src=\"https://shorturl.at/jntJT\" className=\"avatar\" />\n",
            "              <div className=\"list-group-item-content\">\n",
            "                <h3 className=\"title\">Lorem ipsum</h3>\n",
            "                <p className=\"description\">Sample description</p>\n",
            "              </div>\n",
            "            </div>\n",
            "            <div className=\"list-group-item\">\n",
            "              <img src=\"https://shorturl.at/jntJT\" className=\"avatar\" />\n",
            "              <div className=\"list-group-item-content\">\n",
            "                <h3 className=\"title\">Lorem ipsum</h3>\n",
            "                <p className=\"description\">Sample description</p>\n",
            "              </div>\n",
            "            </div>\n",
            "          </div>\n",
            "        </div>\n",
            "      </div>\n",
            "    </div>\n",
            "  );\n",
            "};\n",
            "\n",
            "export default ListGroup;\n",
            "```\n",
            "\n",
            "```css\n",
            ".container {\n",
            "  background-color: #f5f5f5;\n",
            "  padding: 20px;\n",
            "}\n",
            "\n",
            ".row {\n",
            "  margin: 0;\n",
            "}\n",
            "\n",
            ".col-md-4 {\n",
            "  padding: 0 20px;\n",
            "}\n",
            "\n",
            ".list-group {\n",
            "  border: 1px solid #ccc;\n",
            "  border-radius: 5px;\n",
            "  padding: 20px;\n",
            "  height: 300px;\n",
            "  overflow-y: scroll;\n",
            "}\n",
            "\n",
            ".list-group-item {\n",
            "  display: flex;\n",
            "  align-items: center;\n",
            "  margin-bottom: 10px;\n",
            "}\n",
            "\n",
            ".avatar {\n",
            "  width: 50px;\n",
            "  height: 50px;\n",
            "  border-radius: 50%;\n",
            "  margin-right: 10px;\n",
            "}\n",
            "\n",
            ".list-group-item-content {\n",
            "  flex: 1;\n",
            "}\n",
            "\n",
            ".title {\n",
            "  font-size: 18px;\n",
            "  font-weight: bold;\n",
            "  margin-bottom: 5px;\n",
            "}\n",
            "\n",
            ".description {\n",
            "  font-size: 14px;\n",
            "  color: #666;\n",
            "  margin-bottom: 10px;\n",
            "}\n",
            "```\n"
          ]
        }
      ],
      "source": [
        "question = '''\n",
        "Generate a list group component. Have a scrollable container that has atleast 5 lists in it.\n",
        "The list (row-wise) should contain an small rounded avatar (img source : https://shorturl.at/jntJT) and\n",
        "a container (flex-col) that contains a title that says \"Lorem ipsum\" and a description (with accent color) that says \"Sample description\".\n",
        "The title should be aligned with the top of the avatar and the description should align with the the bottom of the avatart\n",
        "\n",
        "Have a light grey as background and make the list white with decent shadows that look good.\n",
        "Makes sure the scrollbar looks good.\n",
        "\n",
        "Give the react component (JS not typescript) as output and follow the styles given in the context.\n",
        "Define all the functions that you use in the component definition.\n",
        "DO NOT USE ANY PROPS instead hardcode the values wherever necessary.\n",
        "\n",
        "ALLOWED : import React from'react';\n",
        "ALLOWED : import \"styles.css\";\n",
        "\n",
        "NO OTHER IMPORTS ARE ALLOWED.\n",
        "FOLLOW THE COLUMN SPANS USING BOOTSTRAP FLEX-BOX OR GRID.\n",
        "GIVE PROPER PADDINGS AND MARGIN WHEREVER NECESSARY SO THAT THE UI LOOKS GOOD.\n",
        "GIVE PROPER FONT-SIZE, FONT-COLOR AND OTHER FONT RELATED THINGS SO THAT THE UI LOOKS GOOD\n",
        "GIVE PROPER COLORS, SHADOWS, ACCENTS, ETC SO THAT THE UI LOOKS GOOD.\n",
        "\n",
        "'''\n",
        "\n",
        "result = rag_chain.invoke(question)\n",
        "\n",
        "print(result['text'])"
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "##### **Timeline**"
      ],
      "metadata": {
        "id": "gz-LmZ_xzq9y"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "question = '''\n",
        "Generate a timeline component. It should be in a vertical fashion. The timeline should be a vertical line that stretches for 75% of the height.\n",
        "Every timestamp should have a big dot on the timeline and should be evenly split across the timeline.\n",
        "While placing the timestamps, start from the top and place them from top to bottom in a evenly spaced manner.\n",
        "There should be 3 timestamps : \"Jan 1, 2024\", \"May 1, 2024\", \"Jul 2, 2024\". These dates should be placed horizontal to their correcsponding dots.\n",
        "These are accent fonts (so have lighter font-color and smaller font-size).\n",
        "Below that, there should be the title (this is the heading but have a small font anyway) that should say \"Title 1\", \"Title 2\" and \"Title 3\".\n",
        "Below the title, there should be a small description that says \" Lorem ipsum is a placeholder text commonly used to demonstrate the visual form of a document or a typeface without relying on meaningful content.\" for each of the timestamps.\n",
        "In the timestamp corresponding to \"Jan 1, 2024\", have a CTA button below the description that says \"Learn More\"\n",
        "\n",
        "Leave some horizontal margin from the timestamp dot to the timestamp content.\n",
        "Make sure the title, description, button, etc (the content of the timestamp) is presented neatly with proper margins and paddings\n",
        "Use a light grey background and proper font colors to make the UI look good.\n",
        "\n",
        "\n",
        "Give the react component (JS not typescript) as output and follow the styles given in the context.\n",
        "Define all the functions that you use in the component definition.\n",
        "DO NOT USE ANY PROPS instead hardcode the values wherever necessary.\n",
        "\n",
        "ALLOWED : import React from'react';\n",
        "ALLOWED : import \"styles.css\";\n",
        "\n",
        "NO OTHER IMPORTS ARE ALLOWED.\n",
        "FOLLOW THE COLUMN SPANS USING BOOTSTRAP FLEX-BOX OR GRID.\n",
        "GIVE PROPER PADDINGS AND MARGIN WHEREVER NECESSARY SO THAT THE UI LOOKS GOOD.\n",
        "GIVE PROPER FONT-SIZE, FONT-COLOR AND OTHER FONT RELATED THINGS SO THAT THE UI LOOKS GOOD\n",
        "GIVE PROPER COLORS, SHADOWS, ACCENTS, ETC SO THAT THE UI LOOKS GOOD.\n",
        "\n",
        "'''\n",
        "\n",
        "result = rag_chain.invoke(question)\n",
        "\n",
        "print(result['text'])"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "LkgJQ08kwTNY",
        "outputId": "36d0e991-98e6-4626-d0d5-a3bd94642e80"
      },
      "execution_count": 21,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\n",
            "```jsx\n",
            "import React from'react';\n",
            "import './styles.css';\n",
            "\n",
            "const Timeline = () => {\n",
            "  return (\n",
            "    <div className=\"timeline-container\">\n",
            "      <div className=\"timeline-line\"></div>\n",
            "      <div className=\"timeline-dots\">\n",
            "        <div className=\"timeline-dot\" style={{ left: '0%' }}>\n",
            "          <div className=\"timeline-content\">\n",
            "            <div className=\"timeline-title\">Title 1</div>\n",
            "            <div className=\"timeline-description\">\n",
            "              Lorem ipsum is a placeholder text commonly used to demonstrate the visual form of a document or a typeface without relying on meaningful content.\n",
            "            </div>\n",
            "          </div>\n",
            "        </div>\n",
            "        <div className=\"timeline-dot\" style={{ left: '25%' }}>\n",
            "          <div className=\"timeline-content\">\n",
            "            <div className=\"timeline-title\">Title 2</div>\n",
            "            <div className=\"timeline-description\">\n",
            "              Lorem ipsum is a placeholder text commonly used to demonstrate the visual form of a document or a typeface without relying on meaningful content.\n",
            "            </div>\n",
            "          </div>\n",
            "        </div>\n",
            "        <div className=\"timeline-dot\" style={{ left: '50%' }}>\n",
            "          <div className=\"timeline-content\">\n",
            "            <div className=\"timeline-title\">Title 3</div>\n",
            "            <div className=\"timeline-description\">\n",
            "              Lorem ipsum is a placeholder text commonly used to demonstrate the visual form of a document or a typeface without relying on meaningful content.\n",
            "            </div>\n",
            "            <div className=\"timeline-button\">\n",
            "              <button>Learn More</button>\n",
            "            </div>\n",
            "          </div>\n",
            "        </div>\n",
            "      </div>\n",
            "    </div>\n",
            "  );\n",
            "};\n",
            "\n",
            "export default Timeline;\n",
            "```\n",
            "\n",
            "```css\n",
            ".timeline-container {\n",
            "  width: 100%;\n",
            "  height: 75vh;\n",
            "  background-color: #f5f5f5;\n",
            "  display: flex;\n",
            "  flex-direction: column;\n",
            "  align-items: center;\n",
            "}\n",
            "\n",
            ".timeline-line {\n",
            "  width: 100%;\n",
            "  height: 1px;\n",
            "  background-color: #ccc;\n",
            "  margin-top: 20px;\n",
            "}\n",
            "\n",
            ".timeline-dots {\n",
            "  display: flex;\n",
            "  flex-direction: row;\n",
            "  align-items: center;\n",
            "  justify-content: space-between;\n",
            "  width: 100%;\n",
            "}\n",
            "\n",
            ".timeline-dot {\n",
            "  width: 10px;\n",
            "  height: 10px;\n",
            "  background-color: #fff;\n",
            "  border-radius: 50%;\n",
            "  margin-right: 10px;\n",
            "  position: relative;\n",
            "}\n",
            "\n",
            ".timeline-content {\n",
            "  display: flex;\n",
            "  flex-direction: column;\n",
            "  align-items: center;\n",
            "  justify-content: center;\n",
            "  width: 100%;\n",
            "  margin-left: 10px;\n",
            "}\n",
            "\n",
            ".timeline-title {\n",
            "  font-size: 16px;\n",
            "  font-weight: bold;\n",
            "  color: #333;\n",
            "  margin-bottom: 10px;\n",
            "}\n",
            "\n",
            ".timeline-description {\n",
            "  font-size: 14px;\n",
            "  color: #666;\n",
            "  margin-bottom: 10px;\n",
            "}\n",
            "\n",
            ".timeline-button {\n",
            "  display: flex;\n",
            "  align-items: center;\n",
            "  justify-content: center;\n",
            "  width: 100%;\n",
            "  margin-top: 10px;\n",
            "}\n",
            "\n",
            ".timeline-button button {\n",
            "  background-color: #007bff;\n",
            "  color: #fff;\n",
            "  font-size: 14px;\n",
            "  padding: 5px 10px;\n",
            "  border: none;\n",
            "  border-radius: 5px;\n",
            "  cursor: pointer;\n",
            "}\n",
            "\n",
            ".timeline-button button:hover {\n",
            "  background-color: #0056b3;\n",
            "}\n",
            "```\n"
          ]
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "The output was terrible. So, let's try providing it some help."
      ],
      "metadata": {
        "id": "qf6EObHe5bVq"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "def get_text_chunks_langchain(text):\n",
        "    text_splitter = CharacterTextSplitter(chunk_size=1024, chunk_overlap=100)\n",
        "    docs = [Document(page_content=x) for x in text_splitter.split_text(text)]\n",
        "    return docs\n",
        "\n",
        "extra_info = '''\n",
        "Basic timeline example\n",
        "Basic example of a responsive vertical timeline, with dates and event names. CSS-only (without JavaScript).\n",
        "\n",
        "<!-- Section: Timeline -->\n",
        "<section class=\"py-5\">\n",
        "  <ul class=\"timeline\">\n",
        "    <li class=\"timeline-item mb-5\">\n",
        "      <h5 class=\"fw-bold\">Our company starts its operations</h5>\n",
        "      <p class=\"text-muted mb-2 fw-bold\">11 March 2020</p>\n",
        "      <p class=\"text-muted\">\n",
        "        Lorem ipsum dolor sit amet consectetur adipisicing elit. Sit\n",
        "        necessitatibus adipisci, ad alias, voluptate pariatur officia\n",
        "        repellendus repellat inventore fugit perferendis totam dolor\n",
        "        voluptas et corrupti distinctio maxime corporis optio?\n",
        "      </p>\n",
        "    </li>\n",
        "\n",
        "    <li class=\"timeline-item mb-5\">\n",
        "      <h5 class=\"fw-bold\">First customer</h5>\n",
        "      <p class=\"text-muted mb-2 fw-bold\">19 March 2020</p>\n",
        "      <p class=\"text-muted\">\n",
        "        Quisque ornare dui nibh, sagittis egestas nisi luctus nec. Sed\n",
        "        aliquet laoreet sapien, eget pulvinar lectus maximus vel.\n",
        "        Phasellus suscipit porta mattis.\n",
        "      </p>\n",
        "    </li>\n",
        "\n",
        "    <li class=\"timeline-item mb-5\">\n",
        "      <h5 class=\"fw-bold\">Our team exceeds 10 people</h5>\n",
        "      <p class=\"text-muted mb-2 fw-bold\">24 June 2020</p>\n",
        "      <p class=\"text-muted\">\n",
        "        Orci varius natoque penatibus et magnis dis parturient montes,\n",
        "        nascetur ridiculus mus. Nulla ullamcorper arcu lacus, maximus\n",
        "        facilisis erat pellentesque nec. Duis et dui maximus dui aliquam\n",
        "        convallis. Quisque consectetur purus erat, et ullamcorper sapien\n",
        "        tincidunt vitae.\n",
        "      </p>\n",
        "    </li>\n",
        "\n",
        "    <li class=\"timeline-item mb-5\">\n",
        "      <h5 class=\"fw-bold\">Earned the first million $!</h5>\n",
        "      <p class=\"text-muted mb-2 fw-bold\">15 October 2020</p>\n",
        "      <p class=\"text-muted\">\n",
        "        Nulla ac tellus convallis, pulvinar nulla ac, fermentum diam. Sed\n",
        "        et urna sit amet massa dapibus tristique non finibus ligula. Nam\n",
        "        pharetra libero nibh, id feugiat tortor rhoncus vitae. Ut suscipit\n",
        "        vulputate mattis.\n",
        "      </p>\n",
        "    </li>\n",
        "  </ul>\n",
        "</section>\n",
        "<!-- Section: Timeline -->\n",
        "\n",
        ".timeline {\n",
        "  border-left: 1px solid hsl(0, 0%, 90%);\n",
        "  position: relative;\n",
        "  list-style: none;\n",
        "}\n",
        "\n",
        ".timeline .timeline-item {\n",
        "  position: relative;\n",
        "}\n",
        "\n",
        ".timeline .timeline-item:after {\n",
        "  position: absolute;\n",
        "  display: block;\n",
        "  top: 0;\n",
        "}\n",
        "\n",
        ".timeline .timeline-item:after {\n",
        "  background-color: hsl(0, 0%, 90%);\n",
        "  left: -38px;\n",
        "  border-radius: 50%;\n",
        "  height: 11px;\n",
        "  width: 11px;\n",
        "  content: \"\";\n",
        "}\n",
        "\n",
        "'''\n",
        "\n",
        "\n",
        "\n",
        "docs = get_text_chunks_langchain(extra_info)\n",
        "\n",
        "# # Load chunked documents into the FAISS index\n",
        "db_extra = FAISS.from_documents(docs, HuggingFaceEmbeddings(model_name='sentence-transformers/all-mpnet-base-v2'))\n",
        "\n",
        "db.merge_from(db_extra)\n",
        "retriever = db.as_retriever()\n",
        "\n",
        "rag_chain = (\n",
        " {\"context\": retriever, \"question\": RunnablePassthrough()}\n",
        "    | llm_chain\n",
        ")"
      ],
      "metadata": {
        "id": "D8KDy7St3BWL"
      },
      "execution_count": 30,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "question = '''\n",
        "Generate a timeline component. It should be in a vertical fashion. The timeline should be a vertical line that stretches for 75% of the height.\n",
        "Every timestamp should have a big dot on the timeline and should be evenly split across the timeline.\n",
        "While placing the timestamps, start from the top and place them from top to bottom in a evenly spaced manner.\n",
        "There should be 3 timestamps : \"Jan 1, 2024\", \"May 1, 2024\", \"Jul 2, 2024\". These dates should be placed horizontal to their correcsponding dots.\n",
        "These are accent fonts (so have lighter font-color and smaller font-size).\n",
        "Below that, there should be the title (this is the heading but have a small font anyway) that should say \"Title 1\", \"Title 2\" and \"Title 3\".\n",
        "Below the title, there should be a small description that says \" Lorem ipsum is a placeholder text commonly used to demonstrate the visual form of a document or a typeface without relying on meaningful content.\" for each of the timestamps.\n",
        "In the timestamp corresponding to \"Jan 1, 2024\", have a CTA button below the description that says \"Learn More\"\n",
        "\n",
        "Leave some horizontal margin from the timestamp dot to the timestamp content.\n",
        "Make sure the title, description, button, etc (the content of the timestamp) is presented neatly with proper margins and paddings\n",
        "Use a light grey background and proper font colors to make the UI look good.\n",
        "\n",
        "\n",
        "Give the react component (JS not typescript) as output and follow the styles given in the context.\n",
        "Define all the functions that you use in the component definition.\n",
        "DO NOT USE ANY PROPS instead hardcode the values wherever necessary.\n",
        "\n",
        "ALLOWED : import React from'react';\n",
        "ALLOWED : import \"styles.css\";\n",
        "\n",
        "NO OTHER IMPORTS ARE ALLOWED.\n",
        "FOLLOW THE COLUMN SPANS USING BOOTSTRAP FLEX-BOX OR GRID.\n",
        "GIVE PROPER PADDINGS AND MARGIN WHEREVER NECESSARY SO THAT THE UI LOOKS GOOD.\n",
        "GIVE PROPER FONT-SIZE, FONT-COLOR AND OTHER FONT RELATED THINGS SO THAT THE UI LOOKS GOOD\n",
        "GIVE PROPER COLORS, SHADOWS, ACCENTS, ETC SO THAT THE UI LOOKS GOOD.\n",
        "\n",
        "'''\n",
        "\n",
        "result = rag_chain.invoke(question)\n",
        "\n",
        "print(result['text'])"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "WDMd1u-55VXo",
        "outputId": "58f24479-c602-4749-e317-3782773cc658"
      },
      "execution_count": 31,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.\n"
          ]
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\n",
            "```jsx\n",
            "import React from'react';\n",
            "import './styles.css';\n",
            "\n",
            "const Timeline = () => {\n",
            "  return (\n",
            "    <div className=\"container\">\n",
            "      <div className=\"row\">\n",
            "        <div className=\"col-md-4\">\n",
            "          <div className=\"timeline\">\n",
            "            <div className=\"timeline-dot\" style={{ backgroundColor: '#ff69b4' }}></div>\n",
            "            <div className=\"timeline-content\">\n",
            "              <div className=\"timeline-title\">Title 1</div>\n",
            "              <div className=\"timeline-description\">\n",
            "                Lorem ipsum is a placeholder text commonly used to demonstrate the visual form of a document or a typeface without relying on meaningful content.\n",
            "              </div>\n",
            "              <div className=\"timeline-button\">\n",
            "                <button className=\"btn btn-primary\">Learn More</button>\n",
            "              </div>\n",
            "            </div>\n",
            "          </div>\n",
            "        </div>\n",
            "        <div className=\"col-md-4\">\n",
            "          <div className=\"timeline\">\n",
            "            <div className=\"timeline-dot\" style={{ backgroundColor: '#008cba' }}></div>\n",
            "            <div className=\"timeline-content\">\n",
            "              <div className=\"timeline-title\">Title 2</div>\n",
            "              <div className=\"timeline-description\">\n",
            "                Lorem ipsum is a placeholder text commonly used to demonstrate the visual form of a document or a typeface without relying on meaningful content.\n",
            "              </div>\n",
            "            </div>\n",
            "          </div>\n",
            "        </div>\n",
            "        <div className=\"col-md-4\">\n",
            "          <div className=\"timeline\">\n",
            "            <div className=\"timeline-dot\" style={{ backgroundColor: '#ffc107' }}></div>\n",
            "            <div className=\"timeline-content\">\n",
            "              <div className=\"timeline-title\">Title 3</div>\n",
            "              <div className=\"timeline-description\">\n",
            "                Lorem ipsum is a placeholder text commonly used to demonstrate the visual form of a document or a typeface without relying on meaningful content.\n",
            "              </div>\n",
            "            </div>\n",
            "          </div>\n",
            "        </div>\n",
            "      </div>\n",
            "    </div>\n",
            "  );\n",
            "};\n",
            "\n",
            "export default Timeline;\n",
            "```\n",
            "\n",
            "```css\n",
            ".timeline {\n",
            "  position: relative;\n",
            "  height: 100%;\n",
            "  background-color: #f5f5f5;\n",
            "  padding: 0;\n",
            "  margin: 0;\n",
            "}\n",
            "\n",
            ".timeline-dot {\n",
            "  position: absolute;\n",
            "  top: 0;\n",
            "  left: 0;\n",
            "  width: 10px;\n",
            "  height: 10px;\n",
            "  border-radius: 50%;\n",
            "  background-color: #fff;\n",
            "  box-shadow: 0 0 2px rgba(0, 0, 0, 0.2);\n",
            "}\n",
            "\n",
            ".timeline-content {\n",
            "  position: absolute;\n",
            "  top: 0;\n",
            "  left: 10px;\n",
            "  width: 100%;\n",
            "  padding: 10px;\n",
            "  background-color: #fff;\n",
            "  box-shadow: 0 0 2px rgba(0, 0, 0, 0.2);\n",
            "}\n",
            "\n",
            ".timeline-title {\n",
            "  font-size: 16px;\n",
            "  font-weight: bold;\n",
            "  margin-bottom: 10px;\n",
            "}\n",
            "\n",
            ".timeline-description {\n",
            "  font-size: 14px;\n",
            "  line-height: 1.5;\n",
            "  margin-bottom: 10px;\n",
            "}\n",
            "\n",
            ".timeline-button {\n",
            "  display: block;\n",
            "  margin: 0 auto;\n",
            "  padding: 5px 10px;\n",
            "  background-color: #007bff;\n",
            "  color: #fff;\n",
            "  border: none;\n",
            "  border-radius: 5px;\n",
            "  text-decoration: none;\n",
            "  font-size: 14px;\n",
            "  font-weight: bold;\n",
            "  cursor: pointer;\n",
            "}\n",
            "\n",
            ".timeline-button:hover {\n",
            "  background-color: #0056b3;\n",
            "}\n",
            "```\n"
          ]
        }
      ]
    }
  ],
  "metadata": {
    "accelerator": "GPU",
    "colab": {
      "collapsed_sections": [
        "KdoxDHTVgyZq",
        "YjH7iwWYg2Rp",
        "1LTISSIIg7_n",
        "ea3d18GVhB4a",
        "mHjUHMMgUx5z",
        "y4t7JCDhU0Mm",
        "GOeW26-QWl9u",
        "PM0pkNgIWuEm",
        "NPV-67fNv1u2",
        "wRK1SXQ3zmqA"
      ],
      "provenance": [],
      "include_colab_link": true
    },
    "kernelspec": {
      "display_name": "Python 3",
      "name": "python3"
    },
    "language_info": {
      "name": "python"
    },
    "widgets": {
      "application/vnd.jupyter.widget-state+json": {
        "af4adf76cdf4419880c741ec94483c7b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_f3be6f78f2aa45edbffb8b821b70da88",
              "IPY_MODEL_492f2b6fee904b25b5b3e28bb868c048",
              "IPY_MODEL_2a19d1089dec4136b21f6efbf9ccef8a"
            ],
            "layout": "IPY_MODEL_e0c241a102344e5fbca4efa4eaf0ebf7"
          }
        },
        "f3be6f78f2aa45edbffb8b821b70da88": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_d38ee4334a474d45a7e63d110ef2f768",
            "placeholder": "​",
            "style": "IPY_MODEL_ecb52d2504f14735b99fdc3aa858aadb",
            "value": "tokenizer_config.json: 100%"
          }
        },
        "492f2b6fee904b25b5b3e28bb868c048": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_36810915b862435299ef5c6425eba7d1",
            "max": 1467,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_a4e484f98d3242969a506cc89f519efa",
            "value": 1467
          }
        },
        "2a19d1089dec4136b21f6efbf9ccef8a": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_a90cc7bc803c4f9c8a835ef301d411ef",
            "placeholder": "​",
            "style": "IPY_MODEL_f24322f9d9b74874b463da99d3a8f611",
            "value": " 1.47k/1.47k [00:00&lt;00:00, 76.0kB/s]"
          }
        },
        "e0c241a102344e5fbca4efa4eaf0ebf7": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "d38ee4334a474d45a7e63d110ef2f768": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "ecb52d2504f14735b99fdc3aa858aadb": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "36810915b862435299ef5c6425eba7d1": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "a4e484f98d3242969a506cc89f519efa": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "a90cc7bc803c4f9c8a835ef301d411ef": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "f24322f9d9b74874b463da99d3a8f611": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "fcdadce3ad594bc99be0118401f3ca8c": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_05a4e2c3a1014351be98e394d9323cd6",
              "IPY_MODEL_cfa3740a580442f887d3e9c9028b9fd1",
              "IPY_MODEL_aeedc5b8b87c4c78b160624c42a5e2cc"
            ],
            "layout": "IPY_MODEL_7f49c57c047b42a29754a46efb82953b"
          }
        },
        "05a4e2c3a1014351be98e394d9323cd6": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_3004ec3921304ab7a1cabc942f5c63cd",
            "placeholder": "​",
            "style": "IPY_MODEL_6a3a252c36b8480880b82d6a1a82aac7",
            "value": "tokenizer.model: 100%"
          }
        },
        "cfa3740a580442f887d3e9c9028b9fd1": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_d21ae7381fd148caba9cffebaf57bb50",
            "max": 493443,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_d9bee2776e2b4d579a422d52985f0fc9",
            "value": 493443
          }
        },
        "aeedc5b8b87c4c78b160624c42a5e2cc": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_54fb446901f14adab581cbe7352f88f2",
            "placeholder": "​",
            "style": "IPY_MODEL_1aecfbec46fc4ee19f764c6913b12637",
            "value": " 493k/493k [00:00&lt;00:00, 7.04MB/s]"
          }
        },
        "7f49c57c047b42a29754a46efb82953b": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "3004ec3921304ab7a1cabc942f5c63cd": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "6a3a252c36b8480880b82d6a1a82aac7": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "d21ae7381fd148caba9cffebaf57bb50": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "d9bee2776e2b4d579a422d52985f0fc9": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "54fb446901f14adab581cbe7352f88f2": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "1aecfbec46fc4ee19f764c6913b12637": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "d6ce92fa8be7429bb4677a0d7c398e39": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_15bf5ca842694c47aeeec35f18e9c281",
              "IPY_MODEL_2d7717ccdca44fca9963d92e6bd23bc0",
              "IPY_MODEL_448214db35414c6ca8af535e7adfb769"
            ],
            "layout": "IPY_MODEL_b1a9e7d315354d93a8aa745e1eef0f70"
          }
        },
        "15bf5ca842694c47aeeec35f18e9c281": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_0ca2fc4aac03495c8ad0b0b465a1ec72",
            "placeholder": "​",
            "style": "IPY_MODEL_9df7c050ab2341aaa3b75d80d3024d7f",
            "value": "tokenizer.json: 100%"
          }
        },
        "2d7717ccdca44fca9963d92e6bd23bc0": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_cc067a51e58743ddbfb318c7d4ad2dcb",
            "max": 1795303,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_181d578663ba4ccf9ed3e628b56b855c",
            "value": 1795303
          }
        },
        "448214db35414c6ca8af535e7adfb769": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_57882ede91ca40549d9fb852c47f0b52",
            "placeholder": "​",
            "style": "IPY_MODEL_0346077d1ad2492f9a1637666638b8c5",
            "value": " 1.80M/1.80M [00:00&lt;00:00, 5.48MB/s]"
          }
        },
        "b1a9e7d315354d93a8aa745e1eef0f70": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "0ca2fc4aac03495c8ad0b0b465a1ec72": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "9df7c050ab2341aaa3b75d80d3024d7f": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "cc067a51e58743ddbfb318c7d4ad2dcb": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "181d578663ba4ccf9ed3e628b56b855c": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "57882ede91ca40549d9fb852c47f0b52": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "0346077d1ad2492f9a1637666638b8c5": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "c20d8ac9c3d248b8992922043968c4c9": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_fc66941a88b64e1a91dce1e1ac2a46e9",
              "IPY_MODEL_704217c1d61a401db13f8ce601343652",
              "IPY_MODEL_0f5f9a8d88fa40d9938777ce5edf99b6"
            ],
            "layout": "IPY_MODEL_55245f68c6fc41b7ab46232028e76b11"
          }
        },
        "fc66941a88b64e1a91dce1e1ac2a46e9": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_667b04eb852d44a7b1b10c401079d1fd",
            "placeholder": "​",
            "style": "IPY_MODEL_0e5aff4822b74cb89496ec679a94b53e",
            "value": "special_tokens_map.json: 100%"
          }
        },
        "704217c1d61a401db13f8ce601343652": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_fbd610be20c04251a0dbf485906f1a87",
            "max": 72,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_b087d738161b484dbbd46f036796d491",
            "value": 72
          }
        },
        "0f5f9a8d88fa40d9938777ce5edf99b6": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_ba0c18a32165483cbe74d79b8c609b1e",
            "placeholder": "​",
            "style": "IPY_MODEL_a062777df055498a8cab521f6d017784",
            "value": " 72.0/72.0 [00:00&lt;00:00, 5.53kB/s]"
          }
        },
        "55245f68c6fc41b7ab46232028e76b11": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "667b04eb852d44a7b1b10c401079d1fd": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "0e5aff4822b74cb89496ec679a94b53e": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "fbd610be20c04251a0dbf485906f1a87": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "b087d738161b484dbbd46f036796d491": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "ba0c18a32165483cbe74d79b8c609b1e": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "a062777df055498a8cab521f6d017784": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "25b55daba0fc44278df2da684ea50efe": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_f6b6e77318cc46e193482ca80b2cff14",
              "IPY_MODEL_9e0afa6274494466baf0cec0c733d3f7",
              "IPY_MODEL_906d975b25d74db8b5fc7ddaee6cec39"
            ],
            "layout": "IPY_MODEL_2eaa234abbec4037b1f0585af2e199e4"
          }
        },
        "f6b6e77318cc46e193482ca80b2cff14": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_bd5c3ca800f04abca2dd2f0c13c90732",
            "placeholder": "​",
            "style": "IPY_MODEL_8cddfe2e5e574544a478203a7d3f82cf",
            "value": "config.json: 100%"
          }
        },
        "9e0afa6274494466baf0cec0c733d3f7": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_740c8ea28f1f4d968c7b85182b64d7d6",
            "max": 571,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_58ffe8d973d94409a8f2583d20119c5b",
            "value": 571
          }
        },
        "906d975b25d74db8b5fc7ddaee6cec39": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_c4bbe15133b24b8ea64149dfa0cd2677",
            "placeholder": "​",
            "style": "IPY_MODEL_33f24463d1d44716bd7b2deff9b9982b",
            "value": " 571/571 [00:00&lt;00:00, 41.8kB/s]"
          }
        },
        "2eaa234abbec4037b1f0585af2e199e4": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "bd5c3ca800f04abca2dd2f0c13c90732": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "8cddfe2e5e574544a478203a7d3f82cf": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "740c8ea28f1f4d968c7b85182b64d7d6": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "58ffe8d973d94409a8f2583d20119c5b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "c4bbe15133b24b8ea64149dfa0cd2677": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "33f24463d1d44716bd7b2deff9b9982b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "d754e7b034c84b80bfc648819ba9f211": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_ebe2bd7737b94bb2bc5765b1ac6df47e",
              "IPY_MODEL_f9a298d47faa4d2fa77d80ae18975913",
              "IPY_MODEL_6c83b57669bc4955a5dc9fff5e634ea8"
            ],
            "layout": "IPY_MODEL_7618e702b27640f096d1f90ae3a16c37"
          }
        },
        "ebe2bd7737b94bb2bc5765b1ac6df47e": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_24a0d424a3e748f9b2b49765c045591f",
            "placeholder": "​",
            "style": "IPY_MODEL_7c2916f53e024124a79ccf7eef43f11e",
            "value": "model.safetensors.index.json: 100%"
          }
        },
        "f9a298d47faa4d2fa77d80ae18975913": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_87b118af30bd42e4b86ee92d954b7814",
            "max": 25125,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_ddf50a86c002498f9dd33cce7492265b",
            "value": 25125
          }
        },
        "6c83b57669bc4955a5dc9fff5e634ea8": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_35a6babf025a4159847bcd0fd50cc2f3",
            "placeholder": "​",
            "style": "IPY_MODEL_34d96018ca5d4df191286e0a534dbb8a",
            "value": " 25.1k/25.1k [00:00&lt;00:00, 1.45MB/s]"
          }
        },
        "7618e702b27640f096d1f90ae3a16c37": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "24a0d424a3e748f9b2b49765c045591f": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "7c2916f53e024124a79ccf7eef43f11e": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "87b118af30bd42e4b86ee92d954b7814": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "ddf50a86c002498f9dd33cce7492265b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "35a6babf025a4159847bcd0fd50cc2f3": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "34d96018ca5d4df191286e0a534dbb8a": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "340937ce8e5b471990c4e2106de493cc": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_6d1a3070727c434f965d671a7a26ab62",
              "IPY_MODEL_4cadb2a4cf214bc3949a944806838793",
              "IPY_MODEL_92c63f6964e04408ad64867cc487a15a"
            ],
            "layout": "IPY_MODEL_5cd415c295dc49419517660fa66a8766"
          }
        },
        "6d1a3070727c434f965d671a7a26ab62": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_2bf79ef2222442c988ac7859f3a5fda1",
            "placeholder": "​",
            "style": "IPY_MODEL_6f27cddba4f4402e832805406ddacebd",
            "value": "Downloading shards: 100%"
          }
        },
        "4cadb2a4cf214bc3949a944806838793": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_bf5c218414664f0587cf9422854e6d22",
            "max": 2,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_745a85fc1244456e9cfd3bf55fd8f572",
            "value": 2
          }
        },
        "92c63f6964e04408ad64867cc487a15a": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_a384d91a1ae244299489abe77a59e736",
            "placeholder": "​",
            "style": "IPY_MODEL_a1b21c29cdfc4e4abe823d4cbc999674",
            "value": " 2/2 [02:18&lt;00:00, 65.55s/it]"
          }
        },
        "5cd415c295dc49419517660fa66a8766": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "2bf79ef2222442c988ac7859f3a5fda1": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "6f27cddba4f4402e832805406ddacebd": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "bf5c218414664f0587cf9422854e6d22": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "745a85fc1244456e9cfd3bf55fd8f572": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "a384d91a1ae244299489abe77a59e736": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "a1b21c29cdfc4e4abe823d4cbc999674": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "752e12e6aff74411bd3d4d2ede8f823f": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_f73457158a154b668849e528baba91a4",
              "IPY_MODEL_432827e0203146ec92dc3ce02db294b2",
              "IPY_MODEL_3e5b5796603b44dd8f02c05384a2c8e2"
            ],
            "layout": "IPY_MODEL_84b343699ac145a99b582cb4f887e19b"
          }
        },
        "f73457158a154b668849e528baba91a4": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_c40cce2d6bc24f18b4dd1d84f07bd261",
            "placeholder": "​",
            "style": "IPY_MODEL_139c412dfd964a83b26f482bcfe2fbee",
            "value": "model-00001-of-00002.safetensors: 100%"
          }
        },
        "432827e0203146ec92dc3ce02db294b2": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_450670fa5c6f407c9111165cf5185255",
            "max": 9942981696,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_9a360d8e8bf54ac196d010abdf650ca7",
            "value": 9942981696
          }
        },
        "3e5b5796603b44dd8f02c05384a2c8e2": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_da76cf47245d4b93a2ad9b885b9eea88",
            "placeholder": "​",
            "style": "IPY_MODEL_1e955068236f4ba081c0eb9c6ef71b97",
            "value": " 9.94G/9.94G [01:29&lt;00:00, 146MB/s]"
          }
        },
        "84b343699ac145a99b582cb4f887e19b": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "c40cce2d6bc24f18b4dd1d84f07bd261": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "139c412dfd964a83b26f482bcfe2fbee": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "450670fa5c6f407c9111165cf5185255": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "9a360d8e8bf54ac196d010abdf650ca7": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "da76cf47245d4b93a2ad9b885b9eea88": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "1e955068236f4ba081c0eb9c6ef71b97": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "3cd11826135049ad953abe22d271f699": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_73b3a8b7a6854fd9a12cb2b0c61e2547",
              "IPY_MODEL_7c8151b8535f47bdb4984d9f5703befb",
              "IPY_MODEL_2953a51914594978a38ebdd5dec512df"
            ],
            "layout": "IPY_MODEL_b9a25088c4664d3ca0b2c698ae607fc1"
          }
        },
        "73b3a8b7a6854fd9a12cb2b0c61e2547": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_502563c04f2b40e498dfe3275cc7ae17",
            "placeholder": "​",
            "style": "IPY_MODEL_5b4d39ad49d348118bf617fd810fc7cc",
            "value": "model-00002-of-00002.safetensors: 100%"
          }
        },
        "7c8151b8535f47bdb4984d9f5703befb": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_90e8389814c54e44bbc3f318fc44268e",
            "max": 4540516344,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_690e58bff44e4f56b23b08d2c1c29f4f",
            "value": 4540516344
          }
        },
        "2953a51914594978a38ebdd5dec512df": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_7d0fee3a7ab84ffc8b5d12372b17b4bd",
            "placeholder": "​",
            "style": "IPY_MODEL_7ba446b49de94c2c9ca4dd719d24e4f6",
            "value": " 4.54G/4.54G [00:48&lt;00:00, 120MB/s]"
          }
        },
        "b9a25088c4664d3ca0b2c698ae607fc1": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "502563c04f2b40e498dfe3275cc7ae17": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "5b4d39ad49d348118bf617fd810fc7cc": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "90e8389814c54e44bbc3f318fc44268e": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "690e58bff44e4f56b23b08d2c1c29f4f": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "7d0fee3a7ab84ffc8b5d12372b17b4bd": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "7ba446b49de94c2c9ca4dd719d24e4f6": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "89fc37ff5bc14ff788bfa52e0feee50b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_cfe9466f34964556907f2f25ed2baf85",
              "IPY_MODEL_acaad782b1864010a15e61e0f3b5f88b",
              "IPY_MODEL_b9e3a96da0cd42d58ede21665f78fa96"
            ],
            "layout": "IPY_MODEL_bff09711a14545d3a90a2f6df24eaaac"
          }
        },
        "cfe9466f34964556907f2f25ed2baf85": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_cd5711916f554825b08f5c7a6a5c2898",
            "placeholder": "​",
            "style": "IPY_MODEL_825750d43e1244a6bf91b58fb387315c",
            "value": "Loading checkpoint shards: 100%"
          }
        },
        "acaad782b1864010a15e61e0f3b5f88b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_c48de966b386410eb8c0af1a99969601",
            "max": 2,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_bb5710405a0e4334ac95b4a9f5a3140d",
            "value": 2
          }
        },
        "b9e3a96da0cd42d58ede21665f78fa96": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_628a796150e147a8941b54ba475f6eac",
            "placeholder": "​",
            "style": "IPY_MODEL_16fa4494aaf748ca954eab6da509eaa6",
            "value": " 2/2 [01:04&lt;00:00, 30.03s/it]"
          }
        },
        "bff09711a14545d3a90a2f6df24eaaac": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "cd5711916f554825b08f5c7a6a5c2898": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "825750d43e1244a6bf91b58fb387315c": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "c48de966b386410eb8c0af1a99969601": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "bb5710405a0e4334ac95b4a9f5a3140d": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "628a796150e147a8941b54ba475f6eac": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "16fa4494aaf748ca954eab6da509eaa6": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "3c05c29e3a3f46109f1845e13138ac81": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_b13eb3c954b74e309072be72835d4567",
              "IPY_MODEL_b88a2b03e32f4ae5aa764bb146441454",
              "IPY_MODEL_4f48e4572c28400c827f8932ef5c9dba"
            ],
            "layout": "IPY_MODEL_212e495e7b774b529cee9e97b9e5be96"
          }
        },
        "b13eb3c954b74e309072be72835d4567": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_40ec30274d1c43d49468d98737b0db4e",
            "placeholder": "​",
            "style": "IPY_MODEL_08f26b7def6847f98ed8d640c6625fbc",
            "value": "generation_config.json: 100%"
          }
        },
        "b88a2b03e32f4ae5aa764bb146441454": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_39f8c47fc8e64ff7a9b1ae4c14ade7f9",
            "max": 116,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_42a634dadd4a4285b11c4c05ca54958f",
            "value": 116
          }
        },
        "4f48e4572c28400c827f8932ef5c9dba": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_c945a280aaa14eeb92e513f87fb65074",
            "placeholder": "​",
            "style": "IPY_MODEL_a5573b2694264856ac90ace744050714",
            "value": " 116/116 [00:00&lt;00:00, 6.12kB/s]"
          }
        },
        "212e495e7b774b529cee9e97b9e5be96": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "40ec30274d1c43d49468d98737b0db4e": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "08f26b7def6847f98ed8d640c6625fbc": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "39f8c47fc8e64ff7a9b1ae4c14ade7f9": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "42a634dadd4a4285b11c4c05ca54958f": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "c945a280aaa14eeb92e513f87fb65074": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "a5573b2694264856ac90ace744050714": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        }
      }
    }
  },
  "nbformat": 4,
  "nbformat_minor": 0
}