Update as_safetensors+fp16.ipynb
Browse files
as_safetensors+fp16.ipynb
CHANGED
@@ -155,7 +155,7 @@
|
|
155 |
"save_type = \".safetensors\" #@param [\".safetensors\", \".ckpt\"]\n",
|
156 |
"#@markdown 変換したモデルをHugging Faceに投稿する場合は「yourname/yourrepo」の形式で投稿先リポジトリを指定<br>\n",
|
157 |
"#@markdown 投稿しない場合は何も入力しない<br>\n",
|
158 |
-
"
|
159 |
"repo_id = \"\" #@param {type:\"string\"}\n",
|
160 |
"\n",
|
161 |
"def upload_to_hugging_face(file_name):\n",
|
@@ -209,7 +209,7 @@
|
|
209 |
" save_model(weights, saved_model := model_name + save_type)\n",
|
210 |
" if repo_id != \"\":\n",
|
211 |
" if os.path.getsize(saved_model) >= 5*1000*1000*1000:\n",
|
212 |
-
" with Repository(os.path.basename(repo_id), clone_from=repo_id, token=True).commit(commit_message=f\"Upload {saved_model} with huggingface_hub\"):\n",
|
213 |
" save_model(weights, saved_model)\n",
|
214 |
" else:\n",
|
215 |
" upload_to_hugging_face(saved_model)\n",
|
|
|
155 |
"save_type = \".safetensors\" #@param [\".safetensors\", \".ckpt\"]\n",
|
156 |
"#@markdown 変換したモデルをHugging Faceに投稿する場合は「yourname/yourrepo」の形式で投稿先リポジトリを指定<br>\n",
|
157 |
"#@markdown 投稿しない場合は何も入力しない<br>\n",
|
158 |
+
"# 5GB以上のファイルを投稿する場合は、投稿先リポジトリを丸ごとダウンロードする工程が挟まるので、時間がかかる場合があります\n",
|
159 |
"repo_id = \"\" #@param {type:\"string\"}\n",
|
160 |
"\n",
|
161 |
"def upload_to_hugging_face(file_name):\n",
|
|
|
209 |
" save_model(weights, saved_model := model_name + save_type)\n",
|
210 |
" if repo_id != \"\":\n",
|
211 |
" if os.path.getsize(saved_model) >= 5*1000*1000*1000:\n",
|
212 |
+
" with Repository(os.path.basename(repo_id), clone_from=repo_id, skip_lfs_files=True, token=True).commit(commit_message=f\"Upload {saved_model} with huggingface_hub\", blocking=False):\n",
|
213 |
" save_model(weights, saved_model)\n",
|
214 |
" else:\n",
|
215 |
" upload_to_hugging_face(saved_model)\n",
|