pytti-tools = source of truth

Author: dmarx

pytti_5_beta.ipynb

@@ -1,23 +1,4 @@
 {
-  "nbformat": 4,
-  "nbformat_minor": 0,
-  "metadata": {
-    "accelerator": "GPU",
-    "colab": {
-      "name": "pytti 5 beta.ipynb",
-      "private_outputs": true,
-      "provenance": [],
-      "collapsed_sections": [],
-      "toc_visible": true
-    },
-    "kernelspec": {
-      "display_name": "Python 3",
-      "name": "python3"
-    },
-    "language_info": {
-      "name": "python"
-    }
-  },
   "cells": [
     {
       "cell_type": "markdown",
@@ -38,10 +19,12 @@
     },
     {
       "cell_type": "code",
+      "execution_count": null,
       "metadata": {
-        "id": "Z0hRb20yxxsc",
-        "cellView": "form"
+        "cellView": "form",
+        "id": "Z0hRb20yxxsc"
       },
+      "outputs": [],
       "source": [
         "# @title Licensed under the MIT License\n",
         "# Copyleft (c) 2021 Henry Rachootin\n",
@@ -63,9 +46,7 @@
         "# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n",
         "# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN\n",
         "# THE SOFTWARE."
-      ],
-      "execution_count": null,
-      "outputs": []
+      ]
     },
     {
       "cell_type": "markdown",
@@ -292,9 +273,11 @@
     },
     {
       "cell_type": "code",
+      "execution_count": null,
       "metadata": {
         "id": "a9C9tARLzyzq"
       },
+      "outputs": [],
       "source": [
         "#@title 1.1 Mount google drive (optional)\n",
         "#@markdown Mounting your drive is optional but recommended. You can even restore from google randomly\n",
@@ -303,16 +286,16 @@
         "drive.mount('/content/drive', force_remount = True)\n",
         "!mkdir -p /content/drive/MyDrive/pytti_test\n",
         "%cd /content/drive/MyDrive/pytti_test"
-      ],
-      "execution_count": null,
-      "outputs": []
+      ]
     },
     {
       "cell_type": "code",
+      "execution_count": null,
       "metadata": {
-        "id": "RosI5DYxtjh0",
-        "cellView": "form"
+        "cellView": "form",
+        "id": "RosI5DYxtjh0"
       },
+      "outputs": [],
       "source": [
         "#@title 1.2 NVIDIA-SMI (optional)\n",
         "#@markdown View information about your runtime GPU.\n",
@@ -326,16 +309,16 @@
         "else:\n",
         "  !nvidia-smi\n",
         "  !nvidia-smi -i 0 -e 0"
-      ],
-      "execution_count": null,
-      "outputs": []
+      ]
     },
     {
       "cell_type": "code",
+      "execution_count": null,
       "metadata": {
-        "id": "CAGyDOe2o9AE",
-        "cellView": "form"
+        "cellView": "form",
+        "id": "CAGyDOe2o9AE"
       },
+      "outputs": [],
       "source": [
         "#@title 1.3 Install everything else\n",
         "#@markdown Run this cell on a fresh runtime to install the libraries and modules.\n",
@@ -362,10 +345,10 @@
         "  !git clone https://github.com/openai/CLIP.git                    &> /dev/null\n",
         "  !git clone https://github.com/CompVis/taming-transformers.git    &> /dev/null\n",
         "  if not path_exists('./pytti'):\n",
-        "    !git clone --branch p5 https://github.com/sportsracer48/pytti.git        &> /dev/null\n",
+        "    !git clone --branch p5 https://github.com/pytti-tools/pytti-core.git        &> /dev/null\n",
         "  else:\n",
         "    !rm -r pytti\n",
-        "    !git clone --branch p5 https://github.com/sportsracer48/pytti.git\n",
+        "    !git clone --branch p5 https://github.com/pytti-tools/pytti-core.git\n",
         "  !git clone https://github.com/shariqfarooq123/AdaBins.git        &> /dev/null\n",
         "  !git clone https://github.com/zacjiang/GMA.git                   &> /dev/null\n",
         "  !mkdir -p AdaBins/pretrained\n",
@@ -386,9 +369,7 @@
         "elif everything_installed:\n",
         "  from pytti.Notebook import change_tqdm_color\n",
         "  change_tqdm_color()"
-      ],
-      "execution_count": null,
-      "outputs": []
+      ]
     },
     {
       "cell_type": "markdown",
@@ -402,10 +383,12 @@
     },
     {
       "cell_type": "code",
+      "execution_count": null,
       "metadata": {
-        "id": "TiKD7os1pyXW",
-        "cellView": "form"
+        "cellView": "form",
+        "id": "TiKD7os1pyXW"
       },
+      "outputs": [],
       "source": [
         "#@title #2.1 Parameters:\n",
         "#@markdown ---\n",
@@ -610,16 +593,16 @@
         "params = Bunch(define_parameters())\n",
         "print(\"SETTINGS:\")\n",
         "print(json.dumps(params))"
-      ],
-      "execution_count": null,
-      "outputs": []
+      ]
     },
     {
       "cell_type": "code",
+      "execution_count": null,
       "metadata": {
-        "id": "lWlZ2Gocb2fF",
-        "cellView": "form"
+        "cellView": "form",
+        "id": "lWlZ2Gocb2fF"
       },
+      "outputs": [],
       "source": [
         "#@title 2.2 Load settings (optional)\n",
         "#@markdown copy the `SETTINGS:` output from the **Parameters** cell (tripple click to select the whole\n",
@@ -664,16 +647,16 @@
         "\n",
         "if settings != '':\n",
         "  params = load_settings(settings, random_seed)"
-      ],
-      "execution_count": null,
-      "outputs": []
+      ]
     },
     {
       "cell_type": "code",
+      "execution_count": null,
       "metadata": {
-        "id": "hqJ6vY2z3rR8",
-        "cellView": "form"
+        "cellView": "form",
+        "id": "hqJ6vY2z3rR8"
       },
+      "outputs": [],
       "source": [
         "#@title 2.3 Run it!\n",
         "#@markdown pytti is 1000% percent better code than VQLIPSE, so have a look at the code. \n",
@@ -1135,9 +1118,7 @@
         "  raise\n",
         "      \n",
         "#print(prof.key_averages().table(sort_by=\"cuda_time_total\", row_limit=10))"
-      ],
-      "execution_count": null,
-      "outputs": []
+      ]
     },
     {
       "cell_type": "markdown",
@@ -1151,10 +1132,12 @@
     },
     {
       "cell_type": "code",
+      "execution_count": null,
       "metadata": {
-        "id": "RZH-r4yyShnX",
-        "cellView": "form"
+        "cellView": "form",
+        "id": "RZH-r4yyShnX"
       },
+      "outputs": [],
       "source": [
         "#@title 3.1 Render video\n",
         "from os.path import exists as path_exists\n",
@@ -1242,16 +1225,16 @@
         "print(\"Encoding video...\")\n",
         "p.wait()\n",
         "print(\"Video complete.\")"
-      ],
-      "execution_count": null,
-      "outputs": []
+      ]
     },
     {
       "cell_type": "code",
+      "execution_count": null,
       "metadata": {
-        "id": "t3EgqHKrSjZx",
-        "cellView": "form"
+        "cellView": "form",
+        "id": "t3EgqHKrSjZx"
       },
+      "outputs": [],
       "source": [
         "#@title 3.1 Render video (concatenate all runs)\n",
         "from os.path import exists as path_exists\n",
@@ -1336,16 +1319,16 @@
         "print(\"Encoding video...\")\n",
         "p.wait()\n",
         "print(\"Video complete.\")"
-      ],
-      "execution_count": null,
-      "outputs": []
+      ]
     },
     {
       "cell_type": "code",
+      "execution_count": null,
       "metadata": {
-        "id": "-qZ8c_-iZ0QM",
-        "cellView": "form"
+        "cellView": "form",
+        "id": "-qZ8c_-iZ0QM"
       },
+      "outputs": [],
       "source": [
         "#@title 3.2 Download the last exported video\n",
         "from os.path import exists as path_exists\n",
@@ -1390,9 +1373,7 @@
         "    #THIS IS NOT AN ERROR. This is the code that would\n",
         "    #make an error if something were wrong.\n",
         "    raise RuntimeError(f\"WARNING: Drive is not mounted.\\nERROR: video videos/{filename} does not exist.\")"
-      ],
-      "execution_count": null,
-      "outputs": []
+      ]
     },
     {
       "cell_type": "markdown",
@@ -1406,10 +1387,12 @@
     },
     {
       "cell_type": "code",
+      "execution_count": null,
       "metadata": {
-        "id": "vHHfWWqoSz35",
-        "cellView": "form"
+        "cellView": "form",
+        "id": "vHHfWWqoSz35"
       },
+      "outputs": [],
       "source": [
         "#@title batch settings\n",
         "from os.path import exists as path_exists\n",
@@ -1530,9 +1513,26 @@
         "namespace = batch_list[0]['file_namespace']\n",
         "if glob.glob(f'images_out/{namespace}/*.png'):\n",
         "  print(f\"WARNING: images_out/{namespace} contains images. Batch indicies may not match filenames unless restoring.\")"
-      ],
-      "execution_count": null,
-      "outputs": []
+      ]
+    }
+  ],
+  "metadata": {
+    "accelerator": "GPU",
+    "colab": {
+      "collapsed_sections": [],
+      "name": "pytti 5 beta.ipynb",
+      "private_outputs": true,
+      "provenance": [],
+      "toc_visible": true
+    },
+    "kernelspec": {
+      "display_name": "Python 3",
+      "name": "python3"
+    },
+    "language_info": {
+      "name": "python"
     }
-  ]
-}
+  },
+  "nbformat": 4,
+  "nbformat_minor": 0
+}