subaqua commited on
Commit
8cadaf8
1 Parent(s): fd95575

Update as_safetensors+fp16.ipynb

Browse files
Files changed (1) hide show
  1. as_safetensors+fp16.ipynb +2 -2
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
- "#@markdown 5GB以上のファイルを投稿する場合は、投稿先リポジトリを丸ごとダウンロードする工程が挟まるので、時間がかかる場合があります\n",
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",