Merge pull request #23 from alembics/fix-clone-of-pytorch3d-lite-in-ipynb

Apply HostServer's disco.py changes for pytorch3d-lite to the .ipynb

Disco_Diffusion.ipynb

@@ -466,6 +466,7 @@
         "  #gitclone(\"https://github.com/facebookresearch/SLIP.git\")\n",
         "  gitclone(\"https://github.com/crowsonkb/guided-diffusion\")\n",
         "  gitclone(\"https://github.com/assafshocher/ResizeRight.git\")\n",
+        "  gitclone(\"https://github.com/MSFTserver/pytorch3d-lite.git\")\n",
         "  pipie(\"./CLIP\")\n",
         "  pipie(\"./guided-diffusion\")\n",
         "  multipip_res = subprocess.run(['pip', 'install', 'lpips', 'datetime', 'timm', 'ftfy'], stdout=subprocess.PIPE).stdout.decode('utf-8')\n",
@@ -491,19 +492,8 @@
         "import sys\n",
         "import torch\n",
         "\n",
-        "#Install pytorch3d\n",
-        "if is_colab:\n",
-        "  pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
-        "  version_str=\"\".join([\n",
-        "      f\"py3{sys.version_info.minor}_cu\",\n",
-        "      torch.version.cuda.replace(\".\",\"\"),\n",
-        "      f\"_pyt{pyt_version_str}\"\n",
-        "  ])\n",
-        "  multipip_res = subprocess.run(['pip', 'install', 'fvcore', 'iopath'], stdout=subprocess.PIPE).stdout.decode('utf-8')\n",
-        "  print(multipip_res)\n",
-        "  subprocess.run(['pip', 'install', '--no-index', '--no-cache-dir', 'pytorch3d', '-f', f'https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html'], stdout=subprocess.PIPE).stdout.decode('utf-8')\n",
-        "\n",
         "# sys.path.append('./SLIP')\n",
+        "sys.path.append('./pytorch3d-lite')\n",
         "sys.path.append('./ResizeRight')\n",
         "sys.path.append('./MiDaS')\n",
         "from dataclasses import dataclass\n",
@@ -725,7 +715,7 @@
         "\n",
         "# https://gist.github.com/adefossez/0646dbe9ed4005480a2407c62aac8869\n",
         "\n",
-        "import pytorch3d.transforms as p3dT\n",
+        "import py3d_tools as p3dT\n",
         "import disco_xform_utils as dxf\n",
         "\n",
         "def interp(t):\n",