min-dalle-test/min_dalle.ipynb

174 lines
166 KiB
Plaintext
Raw Normal View History

2022-06-28 00:58:17 +00:00
{
"cells": [
{
"cell_type": "markdown",
"metadata": {
2022-07-01 21:34:23 +00:00
"id": "view-in-github",
"colab_type": "text"
2022-06-28 00:58:17 +00:00
},
"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"
2022-06-28 00:58:17 +00:00
]
},
{
"cell_type": "code",
2022-07-01 21:34:23 +00:00
"execution_count": null,
2022-06-28 00:58:17 +00:00
"metadata": {
"cellView": "code",
2022-07-01 21:34:23 +00:00
"id": "ix_xt4X1_6F4"
2022-06-28 00:58:17 +00:00
},
2022-07-01 21:34:23 +00:00
"outputs": [],
2022-06-28 00:58:17 +00:00
"source": [
2022-07-01 21:34:23 +00:00
"! pip install min-dalle"
2022-06-28 00:58:17 +00:00
]
},
{
"cell_type": "markdown",
2022-06-30 15:25:24 +00:00
"metadata": {
"id": "kViq2dMbGDKt"
},
"source": [
"### Load Model"
2022-06-30 15:25:24 +00:00
]
},
{
"cell_type": "code",
2022-06-30 15:25:24 +00:00
"execution_count": 2,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
2022-06-30 15:25:24 +00:00
},
"id": "8W-L2ICFGFup",
2022-07-01 21:34:23 +00:00
"outputId": "ed475d82-77e2-437e-acde-6a3e058a3885"
},
"outputs": [
{
2022-07-01 18:19:35 +00:00
"output_type": "stream",
2022-07-01 21:34:23 +00:00
"name": "stdout",
"text": [
"initializing MinDalleTorch\n",
2022-07-01 21:34:23 +00:00
"downloading tokenizer params\n",
"intializing TextTokenizer\n",
"downloading encoder params\n",
"initializing DalleBartEncoderTorch\n",
2022-07-01 21:34:23 +00:00
"downloading decoder params\n",
"initializing DalleBartDecoderTorch\n",
2022-07-01 21:34:23 +00:00
"downloading detokenizer params\n",
"initializing VQGanDetokenizer\n"
]
}
2022-06-30 15:25:24 +00:00
],
"source": [
2022-07-01 20:02:05 +00:00
"from min_dalle import MinDalleTorch\n",
2022-06-30 15:25:24 +00:00
"\n",
2022-07-01 01:23:15 +00:00
"model = MinDalleTorch(is_mega=True, is_reusable=True)"
]
},
2022-06-28 00:58:17 +00:00
{
"cell_type": "markdown",
2022-06-28 15:01:31 +00:00
"metadata": {
2022-06-28 15:05:59 +00:00
"id": "c52TV1GbBNgS"
},
"source": [
"### Generate an Image"
]
2022-06-28 00:58:17 +00:00
},
{
"cell_type": "code",
"execution_count": 4,
2022-06-28 00:58:17 +00:00
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
2022-07-01 01:23:15 +00:00
"height": 511
2022-06-28 15:05:59 +00:00
},
"id": "nQ0UG05dA4p2",
2022-07-01 21:34:23 +00:00
"outputId": "de43cd92-253c-431e-c8c2-26ccd2631925"
2022-06-28 00:58:17 +00:00
},
"outputs": [
{
2022-07-01 18:19:35 +00:00
"output_type": "stream",
2022-07-01 21:34:23 +00:00
"name": "stdout",
2022-06-28 00:58:17 +00:00
"text": [
"tokenizing text\n",
2022-07-01 01:23:15 +00:00
"['Ġ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",
2022-06-28 00:58:17 +00:00
"encoding text tokens\n",
"sampling image tokens\n",
"detokenizing image\n"
]
},
{
2022-07-01 21:34:23 +00:00
"output_type": "display_data",
2022-06-28 00:58:17 +00:00
"data": {
"text/plain": [
2022-07-01 21:34:23 +00:00
"<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
2022-06-28 00:58:17 +00:00
},
2022-07-01 21:34:23 +00:00
"metadata": {}
2022-06-29 17:55:23 +00:00
},
{
2022-07-01 18:19:35 +00:00
"output_type": "stream",
2022-07-01 21:34:23 +00:00
"name": "stdout",
2022-06-29 17:55:23 +00:00
"text": [
2022-07-01 21:34:23 +00:00
"CPU times: user 7.39 s, sys: 4.41 ms, total: 7.39 s\n",
"Wall time: 7.35 s\n"
2022-06-29 17:55:23 +00:00
]
2022-06-28 00:58:17 +00:00
}
2022-06-28 15:05:59 +00:00
],
"source": [
2022-06-29 17:55:23 +00:00
"%%time\n",
"\n",
2022-07-01 01:23:15 +00:00
"text = \"cctv of yoda robbing a liquor store\" #@param {type:\"string\"}\n",
2022-07-01 18:19:35 +00:00
"seed = 0 #@param {type:\"integer\"}\n",
2022-06-28 15:05:59 +00:00
"\n",
2022-07-01 01:23:15 +00:00
"display(model.generate_image(text, seed))"
2022-06-28 00:58:17 +00:00
]
}
],
"metadata": {
"accelerator": "GPU",
"colab": {
2022-06-28 15:05:59 +00:00
"collapsed_sections": [
"Zl_ZFisFApeh"
],
2022-06-30 15:25:24 +00:00
"name": "min-dalle",
2022-07-01 21:34:23 +00:00
"provenance": [],
"include_colab_link": true
2022-06-28 00:58:17 +00:00
},
"gpuClass": "standard",
"kernelspec": {
"display_name": "Python 3",
"name": "python3"
},
"language_info": {
"name": "python"
}
},
"nbformat": 4,
"nbformat_minor": 0
2022-07-01 21:34:23 +00:00
}