Update Fooocus_onColabwithMM.ipynb

Fooocus_onColabwithMM.ipynb

@@ -4,7 +4,8 @@
   "metadata": {
     "colab": {
       "provenance": [],
-      "gpuType": "T4"
+      "gpuType": "T4",
+      "include_colab_link": true
     },
     "kernelspec": {
       "name": "python3",
@@ -16,6 +17,16 @@
     "accelerator": "GPU"
   },
   "cells": [
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "view-in-github",
+        "colab_type": "text"
+      },
+      "source": [
+        "<a href=\"https://colab.research.google.com/github/aquapathos/FooocusJAni/blob/main/Fooocus_onColabwithMM.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+      ]
+    },
     {
       "cell_type": "markdown",
       "source": [
@@ -24,7 +35,8 @@
         "\n",
         "----\n",
         "- とにかく試したいなら「ランタイム」メニューで「すべてのセルを実行」を選び、２の下部の「<font color='darkred'>**約５分かかります**</font>」に進んでください。\n",
-        "- GPUが割り当てられないと実行できません。「GPUバックエンドに接続できません」と表示された場合は後日お試しください。\n",
+        "- 無料ユーザの場合、GPUが割り当てられないことがあります。「<font color='darkblue'>**GPUバックエンドに接続できません**</font>」と表示された場合は、GPUが使えません。その場合、制限はありますが、CPUだけで動作するようにしてあります。\n",
+        "- CPUだけの場合、相当時間がかかります（体感で10倍ぐらい）。メモリも不足するため、速度重視で質を落とす設定となっております。本来の画像品質は得られなことをご承知ください。\n",
         "----\n",
         "# Foooocus + MM LoRA\n",
         "\n",
@@ -44,6 +56,7 @@
         "2. アニメ用のモデルを、animaPencilXL_v310.safetensors から　animagine-xl-3.0.safetensors　に変更\n",
         "3. MM(Mikako Method)のプリセットを追加\n",
         "4. 日本語化リソースの強化\n",
+        "5. CPU Only対応　（ただし10倍ほど時間がかかるので実用性は低い）\n",
         "\n",
         "# １．画像出力フォルダの設定\n",
         "\n",
@@ -68,6 +81,16 @@
     {
       "cell_type": "code",
       "source": [
+        "# GPU のチェック\n",
+        "import torch\n",
+        "# Check for GPU availability\n",
+        "GPU_ready = torch.cuda.is_available()\n",
+        "if GPU_ready:\n",
+        "  print(\"GPU が使えます\")\n",
+        "else:\n",
+        "  print(\"GPU が割り当てられていないため CPUのみで実行します\")\n",
+        "  print(\"非常に時間がかかりますがご了承ください\")\n",
+        "\n",
         "# @markdown #保存フォルダの選択\n",
         "from google.colab import drive\n",
         "import os\n",
@@ -105,20 +128,23 @@
         "\n",
         "# 日本語パッチを追加\n",
         "ja_json_url = \"https://raw.githubusercontent.com/aquapathos/FooocusJAni/refs/heads/main/ja.json\"\n",
+        "\n",
         "ja_json_path = \"/content/Fooocus/language/ja.json\"\n",
         "os.makedirs(os.path.dirname(ja_json_path), exist_ok=True)\n",
         "!curl -L -o {ja_json_path} {ja_json_url}\n",
         "print(f\"ja.json downloaded and saved to {ja_json_path}\")\n",
         "\n",
         "# MM用のプリセットを追加\n",
-        "mm_json_url = \"https://huggingface.co/iciromaco/mm/raw/main/mm_preset.json\"\n",
+        "mm_json_fname = \"mm_preset.json\" if GPU_ready else \"mm_cpu_preset.json\"\n",
+        "mm_json_url = f\"https://raw.githubusercontent.com/aquapathos/FooocusJAni/refs/heads/main/{mm_json_fname}\"\n",
         "mm_json_path = \"/content/Fooocus/presets/mm_preset.json\"\n",
         "os.makedirs(os.path.dirname(mm_json_path), exist_ok=True)\n",
         "!curl -L -o {mm_json_path} {mm_json_url}\n",
         "print(f\"mm_preset.json downloaded and saved to {mm_json_path}\")\n",
         "\n",
         "# アニメ版の設定ファイルのパッチ(最新版)を追加\n",
-        "anime_json_url = \"https://raw.githubusercontent.com/aquapathos/FooocusJAni/refs/heads/main/anime.json\"\n",
+        "anime_json_fname = \"anime.json\" if GPU_ready else \"anime_cpu.json\"\n",
+        "anime_json_url = f\"https://raw.githubusercontent.com/aquapathos/FooocusJAni/refs/heads/main/{anime_json_fname}\"\n",
         "anime_json_path = \"/content/Fooocus/presets/anime.json\"\n",
         "os.makedirs(os.path.dirname(anime_json_path), exist_ok=True)\n",
         "!curl -L -o {anime_json_path} {anime_json_url}\n",
@@ -127,7 +153,8 @@
         "setup_finished = True"
       ],
       "metadata": {
-        "id": "L6hh5BedZ0VJ"
+        "id": "L6hh5BedZ0VJ",
+        "cellView": "form"
       },
       "execution_count": null,
       "outputs": []
@@ -162,25 +189,45 @@
         "モデルタイプ = \"MM\"  # @param [\"デフォルト\", \"アニメ\",\"MM\"] {allow-input: true}\n",
         "\n",
         "if 翻訳機能:\n",
+        "  print(\"H\")\n",
         "  !git clone https://github.com/AlekPet/Fooocus_Extensions_AlekPet.git\n",
         "  !mv Fooocus_Extensions_AlekPet/prompt_translate/webui.py .\n",
         "  !mv Fooocus_Extensions_AlekPet/prompt_translate/modules/* ./modules\n",
         "  !sed -i.bak \"s/label='Enable translate', value=False/label='Enable translate', value=True/\" modules/module_translate.py\n",
         "  !sed -i.bak 's/\"default_prompt\": \"[^\"]*\"/\"default_prompt\": \"湖に映る逆さ絵の風景、人はいない、小さな田舎の家が建っている、家の前には赤いレトロな車、手前には色とりどりのコスモス畑\"/' presets/mm_preset.json\n",
-        "\n",
-        "# Fooocus JP Script Options\n",
-        "if モデルタイプ == \"デフォルト\" and not 日本語メニュー:\n",
-        "    !python entry_with_update.py  --always-high-vram  --output-path {target_local_dir} --share\n",
-        "elif モデルタイプ == \"デフォルト\" and 日本語メニュー :\n",
-        "    !python entry_with_update.py  --always-high-vram  --output-path {target_local_dir} --share --language ja\n",
-        "elif モデルタイプ == \"アニメ\" and not 日本語メニュー:\n",
-        "    !python entry_with_update.py --preset anime --always-high-vram --output-path {target_local_dir} --share\n",
-        "elif モデルタイプ == \"アニメ\" and 日本語メニュー:\n",
-        "    !python entry_with_update.py --preset anime --always-high-vram --output-path {target_local_dir} --share --language ja\n",
-        "elif モデルタイプ == \"MM\" and not 日本語メニュー:\n",
-        "    !python entry_with_update.py --preset mm_preset --always-high-vram --output-path {target_local_dir} --share\n",
-        "elif モデルタイプ == \"MM\" and 日本語メニュー:\n",
-        "    !python entry_with_update.py --preset mm_preset --always-high-vram --output-path {target_local_dir} --share --language ja"
+        "  if not GPU_ready:\n",
+        "    !sed -i.bak 's/HYPER_SD = 4/HYPER_SD = 3/' modules/flags.py\n",
+        "    !sed -i.bak 's/1664\\*576/832\\*576/' modules/flags.py\n",
+        "\n",
+        "if GPU_ready:\n",
+        "  # GPUが使える場合\n",
+        "  if モデルタイプ == \"デフォルト\" and not 日本語メニュー:\n",
+        "      !python entry_with_update.py  --always-high-vram  --output-path {target_local_dir} --share\n",
+        "  elif モデルタイプ == \"デフォルト\" and 日本語メニュー :\n",
+        "      !python entry_with_update.py  --always-high-vram  --output-path {target_local_dir} --share --language ja\n",
+        "  elif モデルタイプ == \"アニメ\" and not 日本語メニュー:\n",
+        "      !python entry_with_update.py --preset anime --always-high-vram --output-path {target_local_dir} --share\n",
+        "  elif モデルタイプ == \"アニメ\" and 日本語メニュー:\n",
+        "      !python entry_with_update.py --preset anime --always-high-vram --output-path {target_local_dir} --share --language ja\n",
+        "  elif モデルタイプ == \"MM\" and not 日本語メニュー:\n",
+        "      !python entry_with_update.py --preset mm_preset --always-high-vram --output-path {target_local_dir} --share\n",
+        "  elif モデルタイプ == \"MM\" and 日本語メニュー:\n",
+        "      !python entry_with_update.py --preset mm_preset --always-high-vram --output-path {target_local_dir} --share --language ja\n",
+        "else:\n",
+        "  # CPU Only\n",
+        "  # GPUが使える場合\n",
+        "  if モデルタイプ == \"デフォルト\" and not 日本語メニュー:\n",
+        "      !python entry_with_update.py --always-cpu 1 --vae-in-fp16 --unet-in-fp8-e5m2 --all-in-fp16 --attention-split --output-path {target_local_dir} --share\n",
+        "  elif モデルタイプ == \"デフォルト\" and 日本語メニュー :\n",
+        "      !python entry_with_update.py --always-cpu 1 --vae-in-fp16 --unet-in-fp8-e5m2 --all-in-fp16 --attention-split --output-path {target_local_dir} --share --language ja\n",
+        "  elif モデルタイプ == \"アニメ\" and not 日本語メニュー:\n",
+        "      !python entry_with_update.py  --always-cpu 1 --vae-in-fp16 --unet-in-fp8-e5m2 --all-in-fp16 --attention-split --preset anime --output-path {target_local_dir} --share\n",
+        "  elif モデルタイプ == \"アニメ\" and 日本語メニュー:\n",
+        "      !python entry_with_update.py  --always-cpu 1 --vae-in-fp16 --unet-in-fp8-e5m2 --all-in-fp16 --attention-split --preset anime --output-path {target_local_dir} --share --language ja\n",
+        "  elif モデルタイプ == \"MM\" and not 日本語メニュー:\n",
+        "      !python entry_with_update.py --always-cpu 1 --vae-in-fp16 --unet-in-fp8-e5m2 --all-in-fp16 --attention-split --preset mm_preset --always-high-vram --output-path {target_local_dir} --share\n",
+        "  elif モデルタイプ == \"MM\" and 日本語メニュー:\n",
+        "      !python entry_with_update.py --always-cpu 1 --vae-in-fp16 --unet-in-fp8-e5m2 --all-in-fp16 --attention-split --preset mm_preset --output-path {target_local_dir} --share --language ja\n"
       ],
       "metadata": {
         "cellView": "form",