Update as_safetensors+fp16.ipynb

as_safetensors+fp16.ipynb

@@ -115,6 +115,7 @@
         "import urllib.parse\n",
         "import urllib.request\n",
         "import wget\n",
+        "import os\n",
         "\n",
         "models = \"Specify_the_model_in_this_way_if_the_model_is_on_My_Drive.safetensors, https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/main/wd-1-4-anime_e1.ckpt, https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/main/wd-1-4-anime_e1.yaml\" #@param {type:\"string\"}\n",
         "models = [m.strip() for m in models.split(\",\")]\n",
@@ -122,6 +123,7 @@
         "  if 0 < len(urllib.parse.urlparse(model).scheme): # if model is url\n",
         "    wget.download(model)\n",
         "  elif model.endswith((\".ckpt\", \".safetensors\", \".yaml\", \".pt\")):\n",
+        "    os.makedirs("/content/" + model, exist_ok=True)\n",
         "    shutil.copy(\"/content/drive/MyDrive/\" + model, \"/content/\" + model) # get the model from mydrive\n",
         "  else:\n",
         "    print(f\"\\\"{model}\\\"はURLではなく、正しい形式のファイルでもありません\")"