Yoshia's_Notebook_Fooocus.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "private_outputs": true,
      "provenance": [],
      "gpuType": "T4",
      "authorship_tag": "ABX9TyMVz311P3rbcnG2aTbU3fS3",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    },
    "accelerator": "GPU"
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/YKefasu/Colab_Notebook-Fooocus/blob/main/Yoshia's_Notebook_Fooocus.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "<center>\n",
        "<font color=\"#CC3E3E\" size=\"6\">⚠️ <u><b>PERHATIAN</b></u> ⚠️</font></center>\n",
        "\n",
        "> <font color=\"#00CED1\">**INFORMASI**:</font>\n",
        "\n",
        "1. Pastikan Subscribe ke 👉 [youtube.com/@yosiakefas](https://www.youtube.com/@yosiakefas) dan Join [Discord Server](https://discord.gg/j8qmYPAGQh) untuk mendapatkan UPDATE Terbaru.\n",
        "2. Karena adanya [Kebijakan][kebijakan] dan [Pernyataan Resmi][pernyataanresmi] Google Colab yang baru, sebagai pengguna google colab gratis akan dibatasi untuk penggunaan GPU dan CUDA ***(komputasinya dibatasi)***\n",
        "3. Jika akun sudah limit pemakaian google colab akun tersebut akan bisa dipakai lagi dalam waktu 24 jam kedepan.\n",
        "(1 akun google memiliki batas 12 jam pemakiaan)\n",
        "4. Dengan mengganti akun google yang dipakai akan mereset limit pemakaian, tetapi ada batasan jumlah pergantian penggunaan secara wajar (maksimal 3 akun, usahakan jangan lebih)\n",
        "\n",
        "[kebijakan]: https://research.google.com/colaboratory/faq.html#limitations-and-restrictions\n",
        "[pernyataanresmi]: https://twitter.com/thechrisperry/status/1649189902079381505\n",
        "\n",
        "<center><font color=\"#CC3E3E\" size=\"6\">⚠️ <u><b>!! <font color=\"#00CED1\" size=\"6\">WAJIB GPU MODE</font> !!</b></u> ⚠️</font></center>\n",
        "\n",
        "\n",
        "> <font color=\"#CC3E3E\">**PERINGATAN**:</font>\n",
        "\n",
        "1. Jangan pernah sering mengganti akun untuk mengatasi limit pengguaan google colab, karena itu sudah termasuk dalam pelanggaran dalam [kebijakan baru][baru] (usahakan tidak menggunakan lebih dari 3 akun).\n",
        "2. DiKarenakan SYARAT & KETENTUAN Google sewaktu-waktu bisa berubah, ada kemungkinan mempengaruhi Colab ini. Jika terjadi masalah/error harap segerapa Lapor ke Server Discord Terima Kasih 🙏.\n",
        "\n",
        "[baru]:https://research.google.com/colaboratory/intl/id/faq.html\n",
        "\n",
        "</br></br>\n",
        "\n",
        "## **<center>Subscribe Youtube Channel & Join Server Discord Kita 😅</center>**\n",
        "<center><font size=\"12\">👉</font> <a href=\"https://trakteer.id/yosiakefas/tip\" target=\"_blank\"><img id=\"wse-buttons-preview\" src=\"https://cdn.trakteer.id/images/embed/trbtn-blue-5.png\" height=\"60\" style=\"border: 0px; height: 60px;\" alt=\"Trakteer Saya\"></a><font size=\"14\">🙏</font></center>\n",
        "</br>\n",
        "<div align=\"center\">\n",
        " <a href=\"https://www.youtube.com/channel/UCsqTcFMsAlegRsUyfyKe_5A\"><img src=\"https://blogger.googleusercontent.com/img/b/R29vZ2xl/AVvXsEiZxKLHX32VYBBOvxvKUDka6uX0wKYYg46EurNzQBUHjaxd2dSJ82paUaJ1OTzpMMt4R7KUuvsbcOLuAg_9NJ1WaRJev8vmRhWea4xfg0AhjSaX2sp3ZFZGD490AArNICVDqzYvrjTqWbKFHxFLE2AYBVaIrGvkwPTqJCroDTvEb43sNuM5wHHCecd6/s350/subscribe-5408999_640.png\" width=\"310\"/></a>ㅤㅤ\n",
        " <a href=\"https://discord.gg/j8qmYPAGQh\"><img src=\"https://cdn.shopify.com/s/files/1/0562/9329/4253/files/Discord_button_480x480.png\" width=\"350\"/></a></div>\n",
        "\n",
        " ___"
      ],
      "metadata": {
        "id": "tNowCyt1c90o"
      }
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "cellView": "form",
        "id": "8h9fydqfLU0i"
      },
      "outputs": [],
      "source": [
        "#@markdown #STEP 1️⃣: Persiapan\n",
        "import os\n",
        "import time\n",
        "from IPython.display import clear_output, display, HTML\n",
        "from IPython.display import clear_output\n",
        "from datetime import timedelta\n",
        "from subprocess import getoutput\n",
        "from google.colab import drive\n",
        "from google.colab.output import eval_js\n",
        "\n",
        "def check_gpu():\n",
        "    print(\"\\033[96m\")  # Cyan text\n",
        "    print('⌚ Checking GPU...', end='')\n",
        "    output = getoutput('nvidia-smi --query-gpu=gpu_name --format=csv')\n",
        "    if \"name\" in output:\n",
        "        gpu_name = output[5:]\n",
        "        print('\\r✅ Current GPU:', gpu_name, flush=True)\n",
        "    else:\n",
        "        print('\\r\\033[91m❎ ERROR: No GPU detected. Lakukan Step dibawah ini.\\n', flush=True)\n",
        "        # Additional code for displaying an image and message\n",
        "        display(HTML(\"<img src='https://i.ibb.co/HC9KH17/NVIDIA-Share-23-01-02-173037.png' width='800px'/>\"))\n",
        "        print('\\033[91m\\nJika tertulis \"Cannot connect to GPU backend\", berarti telah mencapai batas pemakaian. Silahakn istirahat untuk hari ini...')\n",
        "        display(HTML(\"<center><img src='https://media.tenor.com/RY9NX67klacAAAAi/sad-cute.gif' width='272px'/></center>\"))\n",
        "        time.sleep(5)\n",
        "        from google.colab import runtime\n",
        "        runtime.unassign()\n",
        "\n",
        "def clone_FooocusMRE():\n",
        "    %cd /content\n",
        "    !git clone https://github.com/MoonRide303/Fooocus-MRE\n",
        "    %cd /content/Fooocus-MRE\n",
        "    !git checkout 7b795e5bf5318d93c2f3a3ecccd80947f8ff00d3\n",
        "    clear_output()\n",
        "\n",
        "def install_jemalloc():\n",
        "    !apt -y update -qq\n",
        "    !apt -y install libjemalloc-dev\n",
        "    %env LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libjemalloc.so.2\n",
        "    clear_output()\n",
        "\n",
        "def install_req():\n",
        "    %cd /content/Fooocus-MRE/\n",
        "    !pip install torchsde==0.2.5 einops==0.4.1 transformers==4.30.2 safetensors==0.3.1 accelerate==0.21.0\n",
        "    !pip install pytorch_lightning==1.9.4 omegaconf==2.2.3 gradio==3.39.0 xformers==0.0.20 triton==2.0.0 pygit2==1.12.2\n",
        "    !pip install --use-feature=fast-deps -r requirements_versions.txt\n",
        "    clear_output()\n",
        "\n",
        "def download_models():\n",
        "    !apt -y install -qq aria2\n",
        "    clear_output()\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/sd_xl_base_1.0/resolve/main/sd_xl_base_1.0_0.9vae.safetensors -d /content/Fooocus-MRE/models/checkpoints -o sd_xl_base_1.0_0.9vae.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/sd_xl_refiner_1.0/resolve/main/sd_xl_refiner_1.0_0.9vae.safetensors -d /content/Fooocus-MRE/models/checkpoints -o sd_xl_refiner_1.0_0.9vae.safetensors\n",
        "    !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_offset_example-lora_1.0.safetensors -d /content/Fooocus-MRE/models/loras -o sd_xl_offset_example-lora_1.0.safetensors\n",
        "    clear_output()\n",
        "\n",
        "# Check GPU\n",
        "start_colab = int(time.time()) - 5\n",
        "check_gpu()\n",
        "\n",
        "# Check GPU\n",
        "clone_FooocusMRE()\n",
        "\n",
        "#Install jemalloc\n",
        "install_jemalloc()\n",
        "\n",
        "#Install Requirement\n",
        "install_req()\n",
        "\n",
        "#Download Models\n",
        "download_models()\n",
        "\n",
        "from google.colab import drive\n",
        "drive.mount('/content/drive')\n",
        "\n",
        "clear_output()\n",
        "print(\"✅ Done\", \"success\")"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "#@markdown #STEP 2️⃣: Perlengkapan Models Downloader\n",
        "from IPython.display import clear_output\n",
        "import os\n",
        "import requests\n",
        "from urllib.parse import unquote\n",
        "\n",
        "def download_model(Type, DirectLink_URL):\n",
        "    if Type == \"None\" or not DirectLink_URL:\n",
        "        print(\"Skipping download as no valid Type or DirectLink_URL provided.\")\n",
        "        return\n",
        "\n",
        "    output_path = \"/content/Fooocus-MRE/models/\"\n",
        "\n",
        "    if Type == \"Checkpoint\":\n",
        "        output_path += \"checkpoints/\"\n",
        "    elif Type == \"LoRA\":\n",
        "        output_path += \"Loras/\"\n",
        "    elif Type == \"Textual Inversion\":\n",
        "        output_path += \"embeddings/\"\n",
        "    else:\n",
        "        print(\"Invalid type specified.\")\n",
        "        return\n",
        "\n",
        "    # Create the directories if they don't exist\n",
        "    os.makedirs(output_path, exist_ok=True)\n",
        "\n",
        "    urls = [url.strip() for url in DirectLink_URL.split(\",\")]\n",
        "\n",
        "    for url in urls:\n",
        "        response = requests.get(url, stream=True)\n",
        "\n",
        "        if response.status_code == 200:\n",
        "            content_disposition = response.headers.get('content-disposition')\n",
        "            if content_disposition:\n",
        "                filename = unquote(content_disposition.split('filename=')[1])\n",
        "            else:\n",
        "                filename = unquote(url.split(\"/\")[-1])  # Extracting the filename from the URL\n",
        "\n",
        "            # Remove double quotes and semicolons from the filename\n",
        "            filename = filename.replace('\"', '').replace(';', '')\n",
        "\n",
        "            filename = os.path.join(output_path, filename)  # Modify the filename to include the output path\n",
        "\n",
        "            if not os.path.exists(filename):\n",
        "                print(\"Downloading file:\", filename)\n",
        "                chunk_size = 5242880  # 5 MB\n",
        "                with open(filename, 'wb') as f:\n",
        "                    for chunk in response.iter_content(chunk_size=chunk_size):\n",
        "                        if chunk:\n",
        "                            f.write(chunk)\n",
        "                print(\"File downloaded successfully.\")\n",
        "            else:\n",
        "                print(\"File already exists:\", filename)\n",
        "        else:\n",
        "            print(\"Failed to download the file:\", url)\n",
        "\n",
        "    # Clear the output to keep the notebook clean\n",
        "    clear_output()\n",
        "\n",
        "    # Print the success message\n",
        "    print(\"✅ Done\", \"success\")\n",
        "\n",
        "# Call the function with the first set of parameters\n",
        "Type1 = \"Checkpoint\"  # @param [\"Checkpoint\", \"LoRA\", \"Textual Inversion\"]\n",
        "DirectLink_URL1 = \"\"  # @param {'type': 'string'}\n",
        "download_model(Type1, DirectLink_URL1)\n",
        "\n",
        "# Call the function with the second set of parameters\n",
        "Type2 = \"None\"  # @param [\"None\", \"Checkpoint\", \"LoRA\", \"Textual Inversion\"]\n",
        "DirectLink_URL2 = \"\"  # @param {'type': 'string'}\n",
        "download_model(Type2, DirectLink_URL2)"
      ],
      "metadata": {
        "cellView": "form",
        "id": "xJTLCrE3NJgv"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@markdown #STEP 3️⃣: START\n",
        "%cd /content/Fooocus-MRE\n",
        "%env LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libjemalloc.so.2\n",
        "\n",
        "!git pull\n",
        "!python launch.py --share"
      ],
      "metadata": {
        "cellView": "form",
        "id": "rPYk_rQOL_D9"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#@markdown #STEP 4️⃣: Copy Output Folder/Images to GoogleDrive\n",
        "import os\n",
        "import shutil\n",
        "\n",
        "# Define paths\n",
        "source_folder = '/content/Fooocus-MRE/outputs'\n",
        "destination_folder = '/content/drive/MyDrive/Fooocus-IMG-Output/'\n",
        "\n",
        "# Check if the source folder exists\n",
        "if os.path.exists(source_folder):\n",
        "    # Create the destination folder if it doesn't exist\n",
        "    if not os.path.exists(destination_folder):\n",
        "        os.makedirs(destination_folder)\n",
        "\n",
        "    # Recursively copy files from source subdirectories to destination\n",
        "    for root, dirs, files in os.walk(source_folder):\n",
        "        for filename in files:\n",
        "            source_file_path = os.path.join(root, filename)\n",
        "            rel_path = os.path.relpath(source_file_path, source_folder)\n",
        "            destination_file_path = os.path.join(destination_folder, rel_path)\n",
        "\n",
        "            # Create subdirectories in the destination if they don't exist\n",
        "            destination_dir = os.path.dirname(destination_file_path)\n",
        "            os.makedirs(destination_dir, exist_ok=True)\n",
        "\n",
        "            # Check if the file is \"log.html\"\n",
        "            if filename == \"log.html\":\n",
        "                shutil.copy(source_file_path, destination_file_path)\n",
        "                print(f\"✅ File 'log.html' copied (overwritten).\")\n",
        "            else:\n",
        "                # Check if the file already exists in the destination folder\n",
        "                if not os.path.exists(destination_file_path):\n",
        "                    shutil.copy(source_file_path, destination_file_path)\n",
        "                    print(f\"✅ File '{filename}' copied.\")\n",
        "                else:\n",
        "                    print(f\"File '{filename}' already exists. Skipping.\")\n",
        "\n",
        "    print(\"✅ DONE Copy operation completed\")\n",
        "else:\n",
        "    print(\"Source folder not available.\")"
      ],
      "metadata": {
        "cellView": "form",
        "id": "Rf9wO0D3VZfS"
      },
      "execution_count": null,
      "outputs": []
    }
  ]
}