subaqua commited on
Commit
d60ac14
1 Parent(s): b45b3c1

Update as_safetensors+fp16.ipynb

Browse files
Files changed (1) hide show
  1. as_safetensors+fp16.ipynb +1 -1
as_safetensors+fp16.ipynb CHANGED
@@ -123,7 +123,7 @@
123
  " if 0 < len(urllib.parse.urlparse(model).scheme): # if model is url\n",
124
  " wget.download(model)\n",
125
  " elif model.endswith((\".ckpt\", \".safetensors\", \".yaml\", \".pt\")):\n",
126
- " os.makedirs("/content/" + os.path.dirname(model), exist_ok=True)\n",
127
  " shutil.copy(\"/content/drive/MyDrive/\" + model, \"/content/\" + model) # get the model from mydrive\n",
128
  " else:\n",
129
  " print(f\"\\\"{model}\\\"はURLではなく、正しい形式のファイルでもありません\")"
 
123
  " if 0 < len(urllib.parse.urlparse(model).scheme): # if model is url\n",
124
  " wget.download(model)\n",
125
  " elif model.endswith((\".ckpt\", \".safetensors\", \".yaml\", \".pt\")):\n",
126
+ " os.makedirs("/content/" + os.path.basename(model), exist_ok=True)\n",
127
  " shutil.copy(\"/content/drive/MyDrive/\" + model, \"/content/\" + model) # get the model from mydrive\n",
128
  " else:\n",
129
  " print(f\"\\\"{model}\\\"はURLではなく、正しい形式のファイルでもありません\")"