colab requirements

Author: nessessence

align.py

@@ -199,6 +199,7 @@ def get_file(src, tgt):
     if not osp.exists(ALIGNED_IMAGES_DIR):  os.makedirs(ALIGNED_IMAGES_DIR)
 
     files = os.listdir(RAW_IMAGES_DIR)
+    print(f'total img files {len(files)}')
     with tqdm(total=len(files)) as progress:
 
         def cb(*args):

manipulate_note.ipynb

@@ -4801,21 +4801,7 @@
    "execution_count": null,
    "metadata": {},
    "outputs": [],
-   "source": [
-    "To do:\n",
-    "- colab (optional)\n",
-    "    - check if the model can run on a free gpu\n",
-    "    (if the colab works)\n",
-    "    - upload weight to gdrive\n",
-    "        - ddpm\n",
-    "        - classifier\n",
-    "        - ddpm-latent (for unconditional sample)\n",
-    "    - upload image button\n",
-    "    - a scroll for manipuation amplitude\n",
-    "- friendly explaination \n",
-    "\n",
-    "nex"
-   ]
+   "source": []
   },
   {
    "cell_type": "code",

requirement_for_colab.txt

@@ -0,0 +1,11 @@
+torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2  pytorch-lightning==1.2.2  torchtext==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
+scipy==1.5.4
+numpy==1.19.5
+tqdm
+pytorch-fid==0.2.0
+pandas==1.1.5
+lpips==0.1.4
+lmdb==1.2.1
+ftfy
+regex
+dlib requests