Update as_safetensors+fp16.ipynb
Browse files
as_safetensors+fp16.ipynb
CHANGED
|
@@ -85,7 +85,7 @@
|
|
| 85 |
"for model in models:\n",
|
| 86 |
" if 0 < len(urllib.parse.urlparse(model).scheme): # if model is url\n",
|
| 87 |
" wget.download(model)\n",
|
| 88 |
-
" elif model.endswith((\".ckpt\", \".safetensors\")):\n",
|
| 89 |
" shutil.copy(\"/content/drive/MyDrive/\" + model, \"/content/\" + model) # get the model from mydrive\n",
|
| 90 |
" else:\n",
|
| 91 |
" print(f\"\\\"{model}\\\"はURLではなく、正しい形式のファイルでもありません\")"
|
|
|
|
| 85 |
"for model in models:\n",
|
| 86 |
" if 0 < len(urllib.parse.urlparse(model).scheme): # if model is url\n",
|
| 87 |
" wget.download(model)\n",
|
| 88 |
+
" elif model.endswith((\".ckpt\", \".safetensors\", \".yaml\")):\n",
|
| 89 |
" shutil.copy(\"/content/drive/MyDrive/\" + model, \"/content/\" + model) # get the model from mydrive\n",
|
| 90 |
" else:\n",
|
| 91 |
" print(f\"\\\"{model}\\\"はURLではなく、正しい形式のファイルでもありません\")"
|