min-dalle-test/min_dalle.ipynb

{
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/kuprel/min-dalle/blob/main/min_dalle.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "3WL-G_f2_ld8"
      },
      "source": [
        "# min(DALL·E)"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "Zl_ZFisFApeh"
      },
      "source": [
        "### Download models"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "cellView": "code",
        "id": "ix_xt4X1_6F4"
      },
      "outputs": [],
      "source": [
        "! pip install min-dalle"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "kViq2dMbGDKt"
      },
      "source": [
        "### Load Model"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 2,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "8W-L2ICFGFup",
        "outputId": "ed475d82-77e2-437e-acde-6a3e058a3885"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "initializing MinDalleTorch\n",
            "downloading tokenizer params\n",
            "intializing TextTokenizer\n",
            "downloading encoder params\n",
            "initializing DalleBartEncoderTorch\n",
            "downloading decoder params\n",
            "initializing DalleBartDecoderTorch\n",
            "downloading detokenizer params\n",
            "initializing VQGanDetokenizer\n"
          ]
        }
      ],
      "source": [
        "from min_dalle import MinDalleTorch\n",
        "\n",
        "model = MinDalleTorch(is_mega=True, is_reusable=True)"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "c52TV1GbBNgS"
      },
      "source": [
        "### Generate an Image"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 4,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 511
        },
        "id": "nQ0UG05dA4p2",
        "outputId": "de43cd92-253c-431e-c8c2-26ccd2631925"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "tokenizing text\n",
            "['Ġcctv']\n",
            "['Ġof']\n",
            "['Ġyoda']\n",
            "['Ġrob', 'bing']\n",
            "['Ġa']\n",
            "['Ġliquor']\n",
            "['Ġstore']\n",
            "text tokens [0, 17685, 111, 24509, 976, 11811, 58, 13142, 1110, 2]\n",
            "encoding text tokens\n",
            "sampling image tokens\n",
            "detokenizing image\n"
          ]
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "<PIL.Image.Image image mode=RGB size=256x256 at 0x7F9624F58810>"
            ],
            "image/png": "iVBORw0KGgoAAAANSUhEUgAAAQAAAAEACAIAAADTED8xAAEAAElEQVR4nHT9x+9t25Yeho2ZVk47p1/+nXzOzS9U1atMFkEKlmjZsiEDNtx2y/+BG24YtptuWID66hCGIcmgTVimaBar6oV773s3nnzOL++898p5BjfOK4oW7dVcwNprL+Ab8xvzm98YA330YKYb/r17ZwjR3335crdNfL/38RcPMc3+r//lv7AxpBIAwMCHf/Qnv/gf/if/9Oiwc/Hizf/hf/N/3rfZf/q/+J++fXETJ/sfX38JEAOwP/3pP/hP/5P/+IufPsii6n/3v/3Pvn/9al3c2eb5X/7ZL/7wL/7IM/UsKv+r/+pfPPn07Kc/++jf/Dd/+zd//Zv1ZqWbxhc/++Txw4O/+de//OblLwEkAAD0/+lf/dX/6n/9v1Si+T/97//zb19e/MU/+dO//Ks/+82vfvjq69c//8XPGDM2y/nRwZhRIKT88l999eWvvt42awnlzHt4en7/j//kj7uBncflf/3P/h/KgiefPv0n/9Gfzab2xcXb77/99s3Vxdmjc9nSNOOPnzx1XT+vxeWbW6QaTWe9ToeAunxz/d03P9Zl8x//z//D/+z/+J8PptoPr775/Z/rnHz+08+DzvDB+elk0j08PiEsX76f/8t//m9eXrzK8/jBkwcHh6P3b96/+PbtvUdP+uOZowd+P3jy7LHt8G9/83Vn3Pvn/+W/SMuq0+us1vv9Pg66vS/+4CemrnX63Sop4zTWqPHbr7599/L64OD48Hgabvdv375HgO/fP1VSnp6Piixe3Ny6njM9Gl69vfr6ty9yaADg/tmnn//ks7pGr3+4IaZ+fHLoe+7R8UgU1Xdffc9VighBCNuOoTM1Hftn9w78nvV//7/86y9//buijjSdntw7s5mRJ2lV8M5oqBmubbm6rU+GPYXR7cU2K5KmLfM0/vHH7/IqViBboI/OHh0dHXz2xc/9jt3U9ep68ebNxas3d75r51XK24pQMTvrTUddyzLqNOmMur7n8VIdHs0ePH1YtnKxWAspPdfNquzm9vbb71799pffzdfX8P/nOqCs7/u6QR3PbXjtGw7ztc4gmEzH44Peu5c/CIR++a++/uoq/beP/M/+R3+kav3F1Q/f/rChRNdt05YC2YZGTYIc+uSnjy2NFWVuaZSZRK9UXTfM5ohJhpu+52+7Du1xtREvnr8wqD2eDp6/x4oDQOt0telRZzyZvCzf7OXGHAVwcWf7rD/qWgjVcZHtk+Vq/VQ87JqjwO+nVWE67v37J4/uPzo56H9jv/x79EPPNvuDwNVMoTTmkvOPTj/+7PPA6h8dn3SGI11zCNOms15dVcvrK9uU3dGg07dl3AvzfTB1CYMsj87OpoZuMs8oZFlmJZXqeHo+v5sz0xiOJ7bVGfcPdTuoCx7oE9tBu7v88uI1Fw0v5c9/9plneEkRv/jy1Te/+SHPmnudg0n/4d32FSB/dnqCGEaYS2gG0xFAaxCzrFujY7p7jxLCK8kQmU5mZcVbJrMkG97rDfoWz3def/bZp58IrE5O7xGGm4rs1mmZtx2PxvN0B+rqzSKLc40xyijnKJh2x6fT8cHQ8Y20TAxm/+Iv/tA3ddvRXr16FcWhQhiAjQbTg2H+an0BEC/ultH9uNsbPvjoiCMKBilRHdWlZxn3PjpZJXd5Ulg6G3b8B49PBsOOTUmaJrppdPtOvckQJVmSGgONWhpw3qiKITMvU8s1sVRSkRZ40XKdGFLGhmZnVayAShC26wyGAy/QP/v0k7xMj48Pj++d9cdvy7Yo87LNi2gXJndVuVzYHvM6jhKJSMVoMhF1s71YMdM9mcx6w0Fv0M+KIntSfvHR+i/+ePHjN69Xi+1uHX77w7clLP/dAEh5G+62LYBxt+xQS/ejA5iePR7NZkeGQaeze7toNzs5vY2+W8a/f4Qqs2prVBEAoEqKLE9hLju9blOVoqkdW+8EZnS5AAyKENey6mafptv9ahHF+6TMJLS2rbfLzYuXv/voo5+4jk+JbDkAgM4QJ61AwGVrGxqXACAMwi1dNDLcRLtGto7HJU6Flg8OzSefzHRN+0f/5K8Cx0ay9QLj77+LdAfm+KBTQCnqdno8HmNrOhsoighlVRnuNklap0WUmAY9Ohic3zvjXJw/u1e/fLXNb5EU/Z77yRcPHIvZLhkcWlWhRlNbc1Ctit7Au/fozNvshES7NLEl3u3y93fbouCr5d3F6+eYyIPDiQIQoJBE23ynbl8ePRhOD0bMIMnXiWmanqubmsF52wqOlCIYIaTpzAw6AdPJbpfyVWk6KvA7Hd9/8erVQt0NB753/9Af2Eq2QdfjEj56eh9hutokl7eG55mn9w5nR/3lavf1989vL+5ANADUsvxhfxhYRt93HcaiydAx/U+ePer37bLJKxGudzeiVaZjdZzuYpNcrpc1xFn9/u5yZujGqD/mBPZRmlf5BrWi42m6xEgArvvDzsefPv7oowembtZJkRaN4dimb+M9402JJNIIJTrOcNnUQmPc1DQkVSNbhaBtm6blOsWW6QghKmgVAIBCgFUjDU0bDDpjNhBCnJ4WCFuL1W3V1p7t5km8Wc1v7653u3XRFJ3Au7tdLOabw8Pxe3rT1vzRR08+t23jSA9cpxi2o+no/NH9P/rjn8RhtJlvn//48eWb9xdvXiznG8Ub2eKYhzkAADQACS/IDizd2c0jz+54llEmsinAMO1e0MnjMAUwAGzdrItaSAEAtKllDXkUZXnB47gSHAhGnutzpZTEUmDT1oxcq5oqirdpGNdFBRJRnQHEWZ5TDWyDEgwtAABwJBk1FKC2IZIiohMAXdOZ33E7HS8KQ9vuHRzOOoHjOMbpvcM/5n9o6tajhw982472uwdP7t8sfnFz/f5gNjAN3XIsLLBuBZPD47ZBbdtWefju+vXz5y9vr5dxmFBMfvKzTx/c/wMv8F3fOjk/fvHyRwDRiCoYOc+ePjI9rSiqP/rzn4X70HU8w9RdzxvOpmVbp1l5u1zFaajEJgrTKCvv7qL1/LpM0tnBaL/eLleL9XLz6t2bu/WPSnvwj/78L8/PZ5touI+jJE0pY5qpSc4pJYCUlEIoITHn0Cou0iwN95u6zbvDnmOa7y7fC4BH9485nJvaUKHW8RxC2L2Hh1XThkXYiEixYjzzHz077wy9SqaIqM16X8VpVVcIY0KQYeiARH/UdTVzPAwmh4M0i+N8dBod8Er2u0PHDBbbcHBzdbtZAsiyzdu2whQo0Rmrk3Rf5Imu45pAyyVgZDvOeDYbj2cEUEYpviIKGsMydI1WrUJIUIaQopgo3rSiqSVlSgkpJBdN29SirY2O1VYgRKtAACgAzEUTFlFW5GkeBkFHgjBsy+v4+3hrevb0YGYxttuu/OfO65co3G33m2i33b1r3283h4KLbRTOd7uKV1Ee2aY9OBx1u/3BcAZMEy3Pi/LTnz1b3C7S3T7eR5v16vry9ofvfri6uguTbfx7DELcVFd3y10SmYaGGTFMXQBhju10Ush54BABLQZWcQUAlDdcN6x1tMwzkSW1ZjDJpWnrSqpWtKLCjol1XauaqmzK7X5dlCVgOhwMLvVh3TQG0zSGNcKqDwHABUJKKF5Wue94nm3+7JOfTQ8mp6eHjx6eahrhrdoup6NeR2f4YDpjjIECKVQYp1WLXb8zGg63d/Ob6zvg6PWTN4Href5ws0kWd+u723WWiS+/+vry8hXAFuOjn/z8i/PzcyHlbrOt8qrOqqbiALBfx0VWSpAdv9/v0KcPslev3jZtG4f1/Ga1T3eb9W6xWPz43cv5ImsEazkv23q13kNRdHtBJ/AoJkVRzW9X33//PQAkSXR0Oj46PsS65lhmXZZtXbdVZZkGUgqUlEpIIaTkSgoFAhEpobmbXwlo9elUfCA1Hce77Y7qu3DvurZl2hWvyqKosjBwTUJ6gasPOr6tM0PD986OsQIukIYY1XTP8S3TKua5Z1mB6yEsCVaWpvc7naePH4MC2+1iRQ9uV0+Th71VYJru4eGBZrht3WLKLGag3siyNMc1l
          },
          "metadata": {}
        },
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "CPU times: user 7.39 s, sys: 4.41 ms, total: 7.39 s\n",
            "Wall time: 7.35 s\n"
          ]
        }
      ],
      "source": [
        "%%time\n",
        "\n",
        "text = \"cctv of yoda robbing a liquor store\" #@param {type:\"string\"}\n",
        "seed = 0 #@param {type:\"integer\"}\n",
        "\n",
        "display(model.generate_image(text, seed))"
      ]
    }
  ],
  "metadata": {
    "accelerator": "GPU",
    "colab": {
      "collapsed_sections": [
        "Zl_ZFisFApeh"
      ],
      "name": "min-dalle",
      "provenance": [],
      "include_colab_link": true
    },
    "gpuClass": "standard",
    "kernelspec": {
      "display_name": "Python 3",
      "name": "python3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 0
}