test

test.ipynb

@@ -30,7 +30,10 @@
         "!ln -s /content/generative-models/scripts/util/detection/p_head_v1.npz /content/scripts/util/detection/p_head_v1.npz\n",
         "!ln -s /content/generative-models/scripts/util/detection/w_head_v1.npz /content/scripts/util/detection/w_head_v1.npz\n",
         "\n",
-        "import sys, os, math, torch, cv2\n",
+        "import sys\n",
+        "sys.path.append(\"generative-models\")\n",
+        "\n",
+        "import os, math, torch, cv2\n",
         "from omegaconf import OmegaConf\n",
         "from glob import glob\n",
         "from pathlib import Path\n",
@@ -43,8 +46,6 @@
         "from torchvision.transforms import functional as TF\n",
         "from sgm.util import instantiate_from_config\n",
         "\n",
-        "sys.path.append(\"generative-models\")\n",
-        "\n",
         "def load_model(config: str, device: str, num_frames: int, num_steps: int):\n",
         "    config = OmegaConf.load(config)\n",
         "    config.model.params.conditioner_config.params.emb_models[0].params.open_clip_embedding_config.params.init_device = device\n",