{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "d21f84c4-c847-4748-823a-7f83c392ef7b",
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Collecting diffusers\n",
      "  Downloading diffusers-0.32.1-py3-none-any.whl.metadata (18 kB)\n",
      "Collecting transformers\n",
      "  Downloading transformers-4.47.1-py3-none-any.whl.metadata (44 kB)\n",
      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m44.1/44.1 kB\u001b[0m \u001b[31m3.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hRequirement already satisfied: torch in /usr/local/lib/python3.10/dist-packages (2.1.0+cu118)\n",
      "Collecting sentencepiece\n",
      "  Downloading sentencepiece-0.2.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (7.7 kB)\n",
      "Collecting protobuf\n",
      "  Downloading protobuf-5.29.3-cp38-abi3-manylinux2014_x86_64.whl.metadata (592 bytes)\n",
      "Collecting peft\n",
      "  Downloading peft-0.14.0-py3-none-any.whl.metadata (13 kB)\n",
      "Collecting accelerate\n",
      "  Downloading accelerate-1.2.1-py3-none-any.whl.metadata (19 kB)\n",
      "Requirement already satisfied: importlib-metadata in /usr/lib/python3/dist-packages (from diffusers) (4.6.4)\n",
      "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from diffusers) (3.9.0)\n",
      "Collecting huggingface-hub>=0.23.2 (from diffusers)\n",
      "  Downloading huggingface_hub-0.27.1-py3-none-any.whl.metadata (13 kB)\n",
      "Requirement already satisfied: numpy in /usr/local/lib/python3.10/dist-packages (from diffusers) (1.24.1)\n",
      "Collecting regex!=2019.12.17 (from diffusers)\n",
      "  Downloading regex-2024.11.6-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (40 kB)\n",
      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m40.5/40.5 kB\u001b[0m \u001b[31m14.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hRequirement already satisfied: requests in /usr/local/lib/python3.10/dist-packages (from diffusers) (2.31.0)\n",
      "Collecting safetensors>=0.3.1 (from diffusers)\n",
      "  Downloading safetensors-0.5.2-cp38-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (3.8 kB)\n",
      "Requirement already satisfied: Pillow in /usr/local/lib/python3.10/dist-packages (from diffusers) (9.3.0)\n",
      "Requirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.10/dist-packages (from transformers) (23.2)\n",
      "Requirement already satisfied: pyyaml>=5.1 in /usr/local/lib/python3.10/dist-packages (from transformers) (6.0.1)\n",
      "Collecting tokenizers<0.22,>=0.21 (from transformers)\n",
      "  Downloading tokenizers-0.21.0-cp39-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (6.7 kB)\n",
      "Collecting tqdm>=4.27 (from transformers)\n",
      "  Downloading tqdm-4.67.1-py3-none-any.whl.metadata (57 kB)\n",
      "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m57.7/57.7 kB\u001b[0m \u001b[31m21.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hRequirement already satisfied: typing-extensions in /usr/local/lib/python3.10/dist-packages (from torch) (4.4.0)\n",
      "Requirement already satisfied: sympy in /usr/local/lib/python3.10/dist-packages (from torch) (1.12)\n",
      "Requirement already satisfied: networkx in /usr/local/lib/python3.10/dist-packages (from torch) (3.0)\n",
      "Requirement already satisfied: jinja2 in /usr/local/lib/python3.10/dist-packages (from torch) (3.1.2)\n",
      "Requirement already satisfied: fsspec in /usr/local/lib/python3.10/dist-packages (from torch) (2023.4.0)\n",
      "Requirement already satisfied: triton==2.1.0 in /usr/local/lib/python3.10/dist-packages (from torch) (2.1.0)\n",
      "Requirement already satisfied: psutil in /usr/local/lib/python3.10/dist-packages (from peft) (5.9.6)\n",
      "Collecting fsspec (from torch)\n",
      "  Downloading fsspec-2024.12.0-py3-none-any.whl.metadata (11 kB)\n",
      "Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/dist-packages (from jinja2->torch) (2.1.2)\n",
      "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests->diffusers) (2.1.1)\n",
      "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests->diffusers) (3.4)\n",
      "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests->diffusers) (1.26.13)\n",
      "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests->diffusers) (2022.12.7)\n",
      "Requirement already satisfied: mpmath>=0.19 in /usr/local/lib/python3.10/dist-packages (from sympy->torch) (1.3.0)\n",
      "Downloading diffusers-0.32.1-py3-none-any.whl (3.2 MB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m3.2/3.2 MB\u001b[0m \u001b[31m149.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading transformers-4.47.1-py3-none-any.whl (10.1 MB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m10.1/10.1 MB\u001b[0m \u001b[31m151.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m \u001b[36m0:00:01\u001b[0m\n",
      "\u001b[?25hDownloading sentencepiece-0.2.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.3 MB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m1.3/1.3 MB\u001b[0m \u001b[31m157.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading protobuf-5.29.3-cp38-abi3-manylinux2014_x86_64.whl (319 kB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m319.7/319.7 kB\u001b[0m \u001b[31m98.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading peft-0.14.0-py3-none-any.whl (374 kB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m374.8/374.8 kB\u001b[0m \u001b[31m102.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading accelerate-1.2.1-py3-none-any.whl (336 kB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m336.4/336.4 kB\u001b[0m \u001b[31m102.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading huggingface_hub-0.27.1-py3-none-any.whl (450 kB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m450.7/450.7 kB\u001b[0m \u001b[31m112.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading fsspec-2024.12.0-py3-none-any.whl (183 kB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m183.9/183.9 kB\u001b[0m \u001b[31m71.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading regex-2024.11.6-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (781 kB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m781.7/781.7 kB\u001b[0m \u001b[31m144.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading safetensors-0.5.2-cp38-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (461 kB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m462.0/462.0 kB\u001b[0m \u001b[31m117.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading tokenizers-0.21.0-cp39-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (3.0 MB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m3.0/3.0 MB\u001b[0m \u001b[31m165.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hDownloading tqdm-4.67.1-py3-none-any.whl (78 kB)\n",
      "\u001b[2K   \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m78.5/78.5 kB\u001b[0m \u001b[31m36.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
      "\u001b[?25hInstalling collected packages: sentencepiece, tqdm, safetensors, regex, protobuf, fsspec, huggingface-hub, tokenizers, diffusers, accelerate, transformers, peft\n",
      "  Attempting uninstall: fsspec\n",
      "    Found existing installation: fsspec 2023.4.0\n",
      "    Uninstalling fsspec-2023.4.0:\n",
      "      Successfully uninstalled fsspec-2023.4.0\n",
      "Successfully installed accelerate-1.2.1 diffusers-0.32.1 fsspec-2024.12.0 huggingface-hub-0.27.1 peft-0.14.0 protobuf-5.29.3 regex-2024.11.6 safetensors-0.5.2 sentencepiece-0.2.0 tokenizers-0.21.0 tqdm-4.67.1 transformers-4.47.1\n",
      "\u001b[33mWARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv\u001b[0m\u001b[33m\n",
      "\u001b[0m\n",
      "\u001b[1m[\u001b[0m\u001b[34;49mnotice\u001b[0m\u001b[1;39;49m]\u001b[0m\u001b[39;49m A new release of pip is available: \u001b[0m\u001b[31;49m23.3.1\u001b[0m\u001b[39;49m -> \u001b[0m\u001b[32;49m24.3.1\u001b[0m\n",
      "\u001b[1m[\u001b[0m\u001b[34;49mnotice\u001b[0m\u001b[1;39;49m]\u001b[0m\u001b[39;49m To update, run: \u001b[0m\u001b[32;49mpython -m pip install --upgrade pip\u001b[0m\n"
     ]
    }
   ],
   "source": [
    "!pip install diffusers transformers torch sentencepiece protobuf peft accelerate"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "d92544f5-8332-4ed9-8ff3-058bda3bd11e",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "The token has not been saved to the git credentials helper. Pass `add_to_git_credential=True` in this function directly or `--add-to-git-credential` if using via `huggingface-cli` if you want to set the git credential as well.\n",
      "Token is valid (permission: fineGrained).\n",
      "The token `nectar` has been saved to /root/.cache/huggingface/stored_tokens\n",
      "Your token has been saved to /root/.cache/huggingface/token\n",
      "Login successful.\n",
      "The current active token is: `nectar`\n"
     ]
    }
   ],
   "source": [
    "!huggingface-cli login --token ..."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "f13b5804-4f8f-45b3-9fcb-4253e9a0c9bc",
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "from PIL import Image\n",
    "from datetime import datetime\n",
    "from tqdm.auto import tqdm\n",
    "\n",
    "def generate_filename(base_name, extension=\".png\"):\n",
    "    timestamp = datetime.now().strftime(\"%Y%m%d_%H%M%S\")\n",
    "    return f\"{base_name}_{timestamp}{extension}\"\n",
    "\n",
    "def save_image(image, directory, base_name=\"image_grid\"):\n",
    "    filename = generate_filename(base_name)\n",
    "    file_path = os.path.join(directory, filename)\n",
    "    image.save(file_path)\n",
    "    print(f\"Image saved as {file_path}\")\n",
    "\n",
    "def image_grid(imgs, rows, cols, save=True, save_dir='generated_images', base_name=\"image_grid\"):\n",
    "    if not os.path.exists(save_dir):\n",
    "        os.makedirs(save_dir)\n",
    "    \n",
    "    # Adjust rows and cols if the image count doesn't match\n",
    "    total_images = len(imgs)\n",
    "    expected_images = rows * cols\n",
    "    if total_images != expected_images:\n",
    "        print(f\"Warning: Expected {expected_images} images, but got {total_images}. Adjusting grid size.\")\n",
    "        cols = min(total_images, cols)\n",
    "        rows = (total_images + cols - 1) // cols  # Ensure all images fit in the grid\n",
    "\n",
    "    w, h = imgs[0].size\n",
    "    grid = Image.new('RGB', size=(cols * w, rows * h))\n",
    "\n",
    "    for i, img in enumerate(imgs):\n",
    "        grid.paste(img, box=(i % cols * w, i // cols * h))\n",
    "\n",
    "    if save and save_dir:\n",
    "        save_image(grid, save_dir, base_name)\n",
    "\n",
    "    return grid\n",
    "\n",
    "\n",
    "def download_image(url):\n",
    "    try:\n",
    "        response = requests.get(url)\n",
    "    except:\n",
    "        return None\n",
    "    return Image.open(BytesIO(response.content)).convert(\"RGB\")\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "8e476bbe-deb5-4c23-a007-4386606e41b4",
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "57594311206b4300a4da9a9a102cd2fa",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "model_index.json:   0%|          | 0.00/536 [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "e20df890dabf459daa7ef53d15889a66",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Fetching 23 files:   0%|          | 0/23 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "f7d51ce5b1bc41a7bc050ebd87fdbe50",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Loading pipeline components...:   0%|          | 0/7 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "You set `add_prefix_space`. The tokenizer needs to be converted from the slow tokenizers\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "830d0846385546c49b7faf51d9cec2ca",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Loading checkpoint shards:   0%|          | 0/2 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "# import torch\n",
    "# from diffusers import FluxPipeline\n",
    "\n",
    "# pipe = FluxPipeline.from_pretrained(\"black-forest-labs/FLUX.1-dev\", cache_dir=\".\", torch_dtype=torch.bfloat16, force_download=True).to(\"cuda\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "376eaa46-0db0-4d7c-8a46-4226badaf639",
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "The cache for model files in Transformers v4.22.0 has been updated. Migrating your old cache. This is a one-time only operation. You can interrupt this and resume the migration later on by calling `transformers.utils.move_cache()`.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "a3bc5e5702f64798a09367d49e2430c8",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "0it [00:00, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "d0508157a6cb4bd282ebb8ac937c6a06",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "model_index.json:   0%|          | 0.00/536 [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "c8cf30a221974a2dbc89a2365420f890",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Fetching 23 files:   0%|          | 0/23 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "34ff8d84d08c4bcfa4bbbc8919406d1b",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "tokenizer/merges.txt:   0%|          | 0.00/525k [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "e3882d33752348f5b427f7ada56e62bd",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "scheduler/scheduler_config.json:   0%|          | 0.00/273 [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "55f165c3e7604ef1a811980f9b2fb6bb",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "(…)t_encoder_2/model.safetensors.index.json:   0%|          | 0.00/19.9k [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "beead06204ac45a8b7491c8b6f0ef763",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "text_encoder/config.json:   0%|          | 0.00/613 [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "67f648521e25407ea48a3cee16f5fe86",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "model.safetensors:   0%|          | 0.00/246M [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "bbac917ce62e47c795dbe5ed1be6e09f",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "tokenizer/special_tokens_map.json:   0%|          | 0.00/588 [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "d6b357f1951f47d9a731ff406353397a",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "text_encoder_2/config.json:   0%|          | 0.00/782 [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "fa47f58e41ff4dbfad41b9f1a0960876",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "tokenizer/tokenizer_config.json:   0%|          | 0.00/705 [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "ebcc69a04e3d4bc891c51fcd0667ba70",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "tokenizer/vocab.json:   0%|          | 0.00/1.06M [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "d708968cc23d432189565f7daad39375",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "model-00001-of-00002.safetensors:   0%|          | 0.00/4.99G [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "fd95c41827ca40458f832d51f7756fa8",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "tokenizer_2/special_tokens_map.json:   0%|          | 0.00/2.54k [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "f0bfedb0f9b84265b8bd18de02d411c8",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "model-00002-of-00002.safetensors:   0%|          | 0.00/4.53G [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "4bdf545c6157435fbe8a801aa389fee9",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "spiece.model:   0%|          | 0.00/792k [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "f27c2bb1d8e549369d05d9f7cbb92971",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "tokenizer_2/tokenizer.json:   0%|          | 0.00/2.42M [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "e1b8efb017c8464b86632c9542fd70a0",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "transformer/config.json:   0%|          | 0.00/378 [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "46cf458c4adf406d9c5a2cfb27902971",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "tokenizer_2/tokenizer_config.json:   0%|          | 0.00/20.8k [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "40736f32d24b4ad38bb7df5db57c9536",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "(…)pytorch_model-00001-of-00003.safetensors:   0%|          | 0.00/9.98G [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "41b16c4865a241a6abb3d89a4c17696d",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "(…)pytorch_model-00002-of-00003.safetensors:   0%|          | 0.00/9.95G [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "95075fed8a1e4e428db8d271746f7dfe",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "(…)ion_pytorch_model.safetensors.index.json:   0%|          | 0.00/121k [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "5c60a31732b4497eb59826dd42cb023d",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "(…)pytorch_model-00003-of-00003.safetensors:   0%|          | 0.00/3.87G [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "f2dace42aaf943f98777fdfac15c0797",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "vae/config.json:   0%|          | 0.00/820 [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "72cf113565b84b55b053d6dd32ac88fc",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "diffusion_pytorch_model.safetensors:   0%|          | 0.00/168M [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "df785e1a4bf5456da2179e57790008a9",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Loading pipeline components...:   0%|          | 0/7 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "You set `add_prefix_space`. The tokenizer needs to be converted from the slow tokenizers\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "d39c297447c14cb1a4639efa26ac6828",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Loading checkpoint shards:   0%|          | 0/2 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "import torch\n",
    "from diffusers import FluxPipeline\n",
    "\n",
    "flux_dev_pipe = FluxPipeline.from_pretrained(\"black-forest-labs/FLUX.1-dev\", cache_dir=\".\", torch_dtype=torch.bfloat16, force_download=True).to(\"cuda\")"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "5e88a8b3-da8d-4090-b52e-c5f22924eb91",
   "metadata": {},
   "source": [
    "### 2500 steps"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "3b059ea0-4e50-46b9-99ba-dc7b8f7d9582",
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "FluxPipeline {\n",
       "  \"_class_name\": \"FluxPipeline\",\n",
       "  \"_diffusers_version\": \"0.32.1\",\n",
       "  \"_name_or_path\": \"black-forest-labs/FLUX.1-dev\",\n",
       "  \"feature_extractor\": [\n",
       "    null,\n",
       "    null\n",
       "  ],\n",
       "  \"image_encoder\": [\n",
       "    null,\n",
       "    null\n",
       "  ],\n",
       "  \"scheduler\": [\n",
       "    \"diffusers\",\n",
       "    \"FlowMatchEulerDiscreteScheduler\"\n",
       "  ],\n",
       "  \"text_encoder\": [\n",
       "    \"transformers\",\n",
       "    \"CLIPTextModel\"\n",
       "  ],\n",
       "  \"text_encoder_2\": [\n",
       "    \"transformers\",\n",
       "    \"T5EncoderModel\"\n",
       "  ],\n",
       "  \"tokenizer\": [\n",
       "    \"transformers\",\n",
       "    \"CLIPTokenizer\"\n",
       "  ],\n",
       "  \"tokenizer_2\": [\n",
       "    \"transformers\",\n",
       "    \"T5TokenizerFast\"\n",
       "  ],\n",
       "  \"transformer\": [\n",
       "    \"diffusers\",\n",
       "    \"FluxTransformer2DModel\"\n",
       "  ],\n",
       "  \"vae\": [\n",
       "    \"diffusers\",\n",
       "    \"AutoencoderKL\"\n",
       "  ]\n",
       "}"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "lora_path = \"/workspace/taylor-rank16-multi-1e-4-constant/checkpoint-2500/pytorch_lora_weights.safetensors\"\n",
    "flux_dev_pipe.load_lora_weights(lora_path)\n",
    "flux_dev_pipe.to(\"cuda\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "f6490810-f598-4b3d-9748-2895dbfc4409",
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 1/10: Close up photo of rwx woman wearing an exquisite white lace dress with intricate floral patterns, a delicate silver necklace, and pearl earrings. Her makeup is minimal, with a natural glow accentuating her features. The studio backdrop is a soft beige gradient, with professional lighting creating gentle shadows and highlights. Her expression is calm and poised, exuding timeless elegance.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "b15de8a62a0f4d9595d4d972e84722a1",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 2/10: Close up photo of rwx woman wearing a sleek black leather jacket over a white turtleneck sweater. Her hair is styled in a modern bob, and she has a piercing gaze that conveys confidence. The backdrop is an urban skyline at dusk, with golden light reflecting off glass buildings. Subtle HDR lighting enhances the depth of her features.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "be0975dc87c647448f1aef01148a83db",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 3/10: Close up photo of rwx woman in a luxurious evening gown of deep crimson velvet, her hair styled into soft waves cascading over one shoulder. She wears diamond earrings and a matching necklace. The backdrop is a dimly lit grand ballroom, with golden chandeliers casting a warm glow. Her expression is one of quiet sophistication.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "2267bcb8446148c9b5eb5e5b4a45e8f0",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 4/10: Close up photo of rwx woman with her hair tied into a tight, elegant bun, wearing a high-tech metallic silver suit with glowing blue lines integrated into the design. The futuristic backdrop features holographic screens and neon lights. Her intense expression suggests intelligence and focus.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "1e35a85110924cf099df9c133fa7c955",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 5/10: Medium shot photo of rwx woman wearing a traditional sari in vibrant shades of gold and red, with intricate embroidery. She has a delicate bindi on her forehead, and her jewelry includes gold jhumkas and a necklace with emerald accents. The backdrop is a palace courtyard at sunset, with warm orange hues highlighting her serene expression.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "444c17c374de4b45a4397f7e6df8ddc4",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 6/10: Medium shot photo of rwx woman dressed in a rugged explorer outfit: a khaki shirt with rolled-up sleeves, a leather hat, and a bandana around her neck. Her expression is determined, with a hint of adventure. The backdrop is a dense jungle, with sunlight streaming through the leaves.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "d5a6938e5bb442b191539f5e4e6500ba",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 7/10: Medium shot photo of rwx woman in a glowing elven robe of silver and white, her ears subtly pointed, and a delicate circlet of intertwined gold leaves adorning her head. The magical forest backdrop features luminous plants and a soft mist illuminated by moonlight. Her expression is wise and ethereal.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "8b1778fa30a242dc9649af2bf47808c7",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 8/10: Full shot photo of rwx woman wearing an elaborate black and gold masquerade mask, her gown shimmering with sequins that catch the light. Her lips are painted a deep red, and her eyes peer through the mask with a mysterious allure. The backdrop is a Venetian-style ballroom filled with masked figures.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "a46513e141c24b9ba994f41524b864bc",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 9/10: Full shot photo of rwx woman in a tailored white blazer over a silk blouse, her hair in a polished low bun. Her accessories include minimalist gold hoop earrings and a smartwatch. The backdrop is a modern corporate office with floor-to-ceiling windows overlooking a bustling cityscape.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "a7af347a813946299b945bde3bfcde48",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 10/10: Full shot photo of rwx woman with flowing red hair adorned with wildflowers, wearing a simple yet elegant white linen dress. Her expression is serene, and her freckles are highlighted by the golden hour light. The backdrop is a lush meadow with tall grass swaying in the gentle breeze.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "1a4b779ea0704ad7b148a65c09945dab",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "prompts = [\n",
    "    \"Close up photo of rwx woman wearing an exquisite white lace dress with intricate floral patterns, a delicate silver necklace, and pearl earrings. Her makeup is minimal, with a natural glow accentuating her features. The studio backdrop is a soft beige gradient, with professional lighting creating gentle shadows and highlights. Her expression is calm and poised, exuding timeless elegance.\",\n",
    "    \"Close up photo of rwx woman wearing a sleek black leather jacket over a white turtleneck sweater. Her hair is styled in a modern bob, and she has a piercing gaze that conveys confidence. The backdrop is an urban skyline at dusk, with golden light reflecting off glass buildings. Subtle HDR lighting enhances the depth of her features.\",\n",
    "    \"Close up photo of rwx woman in a luxurious evening gown of deep crimson velvet, her hair styled into soft waves cascading over one shoulder. She wears diamond earrings and a matching necklace. The backdrop is a dimly lit grand ballroom, with golden chandeliers casting a warm glow. Her expression is one of quiet sophistication.\",\n",
    "    \"Close up photo of rwx woman with her hair tied into a tight, elegant bun, wearing a high-tech metallic silver suit with glowing blue lines integrated into the design. The futuristic backdrop features holographic screens and neon lights. Her intense expression suggests intelligence and focus.\",\n",
    "    \"Medium shot photo of rwx woman wearing a traditional sari in vibrant shades of gold and red, with intricate embroidery. She has a delicate bindi on her forehead, and her jewelry includes gold jhumkas and a necklace with emerald accents. The backdrop is a palace courtyard at sunset, with warm orange hues highlighting her serene expression.\",\n",
    "    \"Medium shot photo of rwx woman dressed in a rugged explorer outfit: a khaki shirt with rolled-up sleeves, a leather hat, and a bandana around her neck. Her expression is determined, with a hint of adventure. The backdrop is a dense jungle, with sunlight streaming through the leaves.\",\n",
    "    \"Medium shot photo of rwx woman in a glowing elven robe of silver and white, her ears subtly pointed, and a delicate circlet of intertwined gold leaves adorning her head. The magical forest backdrop features luminous plants and a soft mist illuminated by moonlight. Her expression is wise and ethereal.\",\n",
    "    \"Full shot photo of rwx woman wearing an elaborate black and gold masquerade mask, her gown shimmering with sequins that catch the light. Her lips are painted a deep red, and her eyes peer through the mask with a mysterious allure. The backdrop is a Venetian-style ballroom filled with masked figures.\",\n",
    "    \"Full shot photo of rwx woman in a tailored white blazer over a silk blouse, her hair in a polished low bun. Her accessories include minimalist gold hoop earrings and a smartwatch. The backdrop is a modern corporate office with floor-to-ceiling windows overlooking a bustling cityscape.\",\n",
    "    \"Full shot photo of rwx woman with flowing red hair adorned with wildflowers, wearing a simple yet elegant white linen dress. Her expression is serene, and her freckles are highlighted by the golden hour light. The backdrop is a lush meadow with tall grass swaying in the gentle breeze.\"\n",
    "]\n",
    "\n",
    "save_dir = \"taylor_lora_2500steps_generated_images\"\n",
    "for i, prompt in enumerate(prompts):\n",
    "    print(f\"Generating images for prompt {i+1}/{len(prompts)}: {prompt}\")\n",
    "    images = flux_dev_pipe(prompt, \n",
    "                  num_inference_steps=35, \n",
    "                  num_images_per_prompt=1, \n",
    "                  width=1024,\n",
    "                  height=1024,\n",
    "                  guidance_scale=3.5,\n",
    "                  generator=torch.Generator(\"cpu\").manual_seed(1024)).images\n",
    "\n",
    "    if not os.path.exists(save_dir):\n",
    "        os.makedirs(save_dir)\n",
    "        \n",
    "    images[0].save(f\"{save_dir}/prompt_{i+1}.jpg\")"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "e577d171-79d7-4a91-8cea-8327acc584df",
   "metadata": {},
   "source": [
    "### 2000 steps"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "1dd352b6-d0b8-41b8-a38d-503cb8db9ea5",
   "metadata": {
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "flux_dev_pipe.fuse_lora(lora_scale=0)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "2f79cdf8-a39a-4a22-93b6-d83fd0182a79",
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/usr/local/lib/python3.10/dist-packages/peft/tuners/lora/model.py:431: UserWarning: Adapter cannot be set when the model is merged. Unmerging the model first.\n",
      "  warnings.warn(\"Adapter cannot be set when the model is merged. Unmerging the model first.\")\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "FluxPipeline {\n",
       "  \"_class_name\": \"FluxPipeline\",\n",
       "  \"_diffusers_version\": \"0.32.1\",\n",
       "  \"_name_or_path\": \"black-forest-labs/FLUX.1-dev\",\n",
       "  \"feature_extractor\": [\n",
       "    null,\n",
       "    null\n",
       "  ],\n",
       "  \"image_encoder\": [\n",
       "    null,\n",
       "    null\n",
       "  ],\n",
       "  \"scheduler\": [\n",
       "    \"diffusers\",\n",
       "    \"FlowMatchEulerDiscreteScheduler\"\n",
       "  ],\n",
       "  \"text_encoder\": [\n",
       "    \"transformers\",\n",
       "    \"CLIPTextModel\"\n",
       "  ],\n",
       "  \"text_encoder_2\": [\n",
       "    \"transformers\",\n",
       "    \"T5EncoderModel\"\n",
       "  ],\n",
       "  \"tokenizer\": [\n",
       "    \"transformers\",\n",
       "    \"CLIPTokenizer\"\n",
       "  ],\n",
       "  \"tokenizer_2\": [\n",
       "    \"transformers\",\n",
       "    \"T5TokenizerFast\"\n",
       "  ],\n",
       "  \"transformer\": [\n",
       "    \"diffusers\",\n",
       "    \"FluxTransformer2DModel\"\n",
       "  ],\n",
       "  \"vae\": [\n",
       "    \"diffusers\",\n",
       "    \"AutoencoderKL\"\n",
       "  ]\n",
       "}"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "lora_path = \"/workspace/taylor-rank16-multi-1e-4-constant/checkpoint-2000/pytorch_lora_weights.safetensors\"\n",
    "flux_dev_pipe.load_lora_weights(lora_path)\n",
    "flux_dev_pipe.to(\"cuda\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "4d14fb44-106c-4225-848f-c79da7c2d725",
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 1/10: Close up photo of ext woman wearing an exquisite white lace dress with intricate floral patterns, a delicate silver necklace, and pearl earrings. Her makeup is minimal, with a natural glow accentuating her features. The studio backdrop is a soft beige gradient, with professional lighting creating gentle shadows and highlights. Her expression is calm and poised, exuding timeless elegance.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "f2e191d21daf4cd9a67b055274b85a60",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 2/10: Close up photo of ext woman wearing a sleek black leather jacket over a white turtleneck sweater. Her hair is styled in a modern bob, and she has a piercing gaze that conveys confidence. The backdrop is an urban skyline at dusk, with golden light reflecting off glass buildings. Subtle HDR lighting enhances the depth of her features.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "60394f15302e486f88f41af5257621c4",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 3/10: Close up photo of ext woman in a luxurious evening gown of deep crimson velvet, her hair styled into soft waves cascading over one shoulder. She wears diamond earrings and a matching necklace. The backdrop is a dimly lit grand ballroom, with golden chandeliers casting a warm glow. Her expression is one of quiet sophistication.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "8600b4419476493f96ef0ed7d87e3663",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 4/10: Close up photo of ext woman with her hair tied into a tight, elegant bun, wearing a high-tech metallic silver suit with glowing blue lines integrated into the design. The futuristic backdrop features holographic screens and neon lights. Her intense expression suggests intelligence and focus.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "045342b2c9684dbe8e184c679eb4cf8c",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 5/10: Medium shot photo of ext woman wearing a traditional sari in vibrant shades of gold and red, with intricate embroidery. She has a delicate bindi on her forehead, and her jewelry includes gold jhumkas and a necklace with emerald accents. The backdrop is a palace courtyard at sunset, with warm orange hues highlighting her serene expression.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "3b546d949fac4659ba3457782e2156a6",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 6/10: Medium shot photo of ext woman dressed in a rugged explorer outfit: a khaki shirt with rolled-up sleeves, a leather hat, and a bandana around her neck. Her expression is determined, with a hint of adventure. The backdrop is a dense jungle, with sunlight streaming through the leaves.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "e6a5f678e3ed41a7a80bfd4c13f5020e",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 7/10: Medium shot photo of ext woman in a glowing elven robe of silver and white, her ears subtly pointed, and a delicate circlet of intertwined gold leaves adorning her head. The magical forest backdrop features luminous plants and a soft mist illuminated by moonlight. Her expression is wise and ethereal.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "cf781fda68084775a7722256f32bebcc",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 8/10: Full shot photo of ext woman wearing an elaborate black and gold masquerade mask, her gown shimmering with sequins that catch the light. Her lips are painted a deep red, and her eyes peer through the mask with a mysterious allure. The backdrop is a Venetian-style ballroom filled with masked figures.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "dd43e6cd2ad3464ea2cc0bd6fd15f7db",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 9/10: Full shot photo of ext woman in a tailored white blazer over a silk blouse, her hair in a polished low bun. Her accessories include minimalist gold hoop earrings and a smartwatch. The backdrop is a modern corporate office with floor-to-ceiling windows overlooking a bustling cityscape.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "740d1fb40b404d07b65aae7844ecb8f3",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 10/10: Full shot photo of ext woman with flowing red hair adorned with wildflowers, wearing a simple yet elegant white linen dress. Her expression is serene, and her freckles are highlighted by the golden hour light. The backdrop is a lush meadow with tall grass swaying in the gentle breeze.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "e13432e9b2ac48ca9d963a0155822203",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "prompts = [\n",
    "    \"Close up photo of ext woman wearing an exquisite white lace dress with intricate floral patterns, a delicate silver necklace, and pearl earrings. Her makeup is minimal, with a natural glow accentuating her features. The studio backdrop is a soft beige gradient, with professional lighting creating gentle shadows and highlights. Her expression is calm and poised, exuding timeless elegance.\",\n",
    "    \"Close up photo of ext woman wearing a sleek black leather jacket over a white turtleneck sweater. Her hair is styled in a modern bob, and she has a piercing gaze that conveys confidence. The backdrop is an urban skyline at dusk, with golden light reflecting off glass buildings. Subtle HDR lighting enhances the depth of her features.\",\n",
    "    \"Close up photo of ext woman in a luxurious evening gown of deep crimson velvet, her hair styled into soft waves cascading over one shoulder. She wears diamond earrings and a matching necklace. The backdrop is a dimly lit grand ballroom, with golden chandeliers casting a warm glow. Her expression is one of quiet sophistication.\",\n",
    "    \"Close up photo of ext woman with her hair tied into a tight, elegant bun, wearing a high-tech metallic silver suit with glowing blue lines integrated into the design. The futuristic backdrop features holographic screens and neon lights. Her intense expression suggests intelligence and focus.\",\n",
    "    \"Medium shot photo of ext woman wearing a traditional sari in vibrant shades of gold and red, with intricate embroidery. She has a delicate bindi on her forehead, and her jewelry includes gold jhumkas and a necklace with emerald accents. The backdrop is a palace courtyard at sunset, with warm orange hues highlighting her serene expression.\",\n",
    "    \"Medium shot photo of ext woman dressed in a rugged explorer outfit: a khaki shirt with rolled-up sleeves, a leather hat, and a bandana around her neck. Her expression is determined, with a hint of adventure. The backdrop is a dense jungle, with sunlight streaming through the leaves.\",\n",
    "    \"Medium shot photo of ext woman in a glowing elven robe of silver and white, her ears subtly pointed, and a delicate circlet of intertwined gold leaves adorning her head. The magical forest backdrop features luminous plants and a soft mist illuminated by moonlight. Her expression is wise and ethereal.\",\n",
    "    \"Full shot photo of ext woman wearing an elaborate black and gold masquerade mask, her gown shimmering with sequins that catch the light. Her lips are painted a deep red, and her eyes peer through the mask with a mysterious allure. The backdrop is a Venetian-style ballroom filled with masked figures.\",\n",
    "    \"Full shot photo of ext woman in a tailored white blazer over a silk blouse, her hair in a polished low bun. Her accessories include minimalist gold hoop earrings and a smartwatch. The backdrop is a modern corporate office with floor-to-ceiling windows overlooking a bustling cityscape.\",\n",
    "    \"Full shot photo of ext woman with flowing red hair adorned with wildflowers, wearing a simple yet elegant white linen dress. Her expression is serene, and her freckles are highlighted by the golden hour light. The backdrop is a lush meadow with tall grass swaying in the gentle breeze.\"\n",
    "]\n",
    "\n",
    "save_dir = \"lora_2000steps_generated_images\"\n",
    "for i, prompt in enumerate(prompts):\n",
    "    print(f\"Generating images for prompt {i+1}/{len(prompts)}: {prompt}\")\n",
    "    images = flux_dev_pipe(prompt, \n",
    "                  num_inference_steps=35, \n",
    "                  num_images_per_prompt=1, \n",
    "                  width=1024,\n",
    "                  height=1024,\n",
    "                  guidance_scale=3.5,\n",
    "                  generator=torch.Generator(\"cpu\").manual_seed(1024)).images\n",
    "\n",
    "    if not os.path.exists(save_dir):\n",
    "        os.makedirs(save_dir)\n",
    "               \n",
    "    images[0].save(f\"{save_dir}/prompt_{i+1}.jpg\")"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "79844ea2-e27a-4043-8f71-1f21ed14f16d",
   "metadata": {},
   "source": [
    "### 1500 steps"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "270fcbfc-72d2-4655-9407-2162c3d16df4",
   "metadata": {},
   "outputs": [],
   "source": [
    "flux_dev_pipe.fuse_lora(lora_scale=0)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "cc6eea5b-8811-455a-b9bb-bff5a5f0326e",
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "FluxPipeline {\n",
       "  \"_class_name\": \"FluxPipeline\",\n",
       "  \"_diffusers_version\": \"0.32.1\",\n",
       "  \"_name_or_path\": \"black-forest-labs/FLUX.1-dev\",\n",
       "  \"feature_extractor\": [\n",
       "    null,\n",
       "    null\n",
       "  ],\n",
       "  \"image_encoder\": [\n",
       "    null,\n",
       "    null\n",
       "  ],\n",
       "  \"scheduler\": [\n",
       "    \"diffusers\",\n",
       "    \"FlowMatchEulerDiscreteScheduler\"\n",
       "  ],\n",
       "  \"text_encoder\": [\n",
       "    \"transformers\",\n",
       "    \"CLIPTextModel\"\n",
       "  ],\n",
       "  \"text_encoder_2\": [\n",
       "    \"transformers\",\n",
       "    \"T5EncoderModel\"\n",
       "  ],\n",
       "  \"tokenizer\": [\n",
       "    \"transformers\",\n",
       "    \"CLIPTokenizer\"\n",
       "  ],\n",
       "  \"tokenizer_2\": [\n",
       "    \"transformers\",\n",
       "    \"T5TokenizerFast\"\n",
       "  ],\n",
       "  \"transformer\": [\n",
       "    \"diffusers\",\n",
       "    \"FluxTransformer2DModel\"\n",
       "  ],\n",
       "  \"vae\": [\n",
       "    \"diffusers\",\n",
       "    \"AutoencoderKL\"\n",
       "  ]\n",
       "}"
      ]
     },
     "execution_count": 11,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "lora_path = \"/workspace/taylor-rank16-multi-1e-4-constant/checkpoint-1500/pytorch_lora_weights.safetensors\"\n",
    "flux_dev_pipe.load_lora_weights(lora_path)\n",
    "flux_dev_pipe.to(\"cuda\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "9dfb7f4c-aeaf-4ec8-a227-fbecded489c3",
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 1/10: Close up photo of ext woman wearing an exquisite white lace dress with intricate floral patterns, a delicate silver necklace, and pearl earrings. Her makeup is minimal, with a natural glow accentuating her features. The studio backdrop is a soft beige gradient, with professional lighting creating gentle shadows and highlights. Her expression is calm and poised, exuding timeless elegance.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "33b2a3017a63415e94020620e969f868",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 2/10: Close up photo of ext woman wearing a sleek black leather jacket over a white turtleneck sweater. Her hair is styled in a modern bob, and she has a piercing gaze that conveys confidence. The backdrop is an urban skyline at dusk, with golden light reflecting off glass buildings. Subtle HDR lighting enhances the depth of her features.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "62910d8d8bee4b1bb0ed5dcc8d9ce933",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 3/10: Close up photo of ext woman in a luxurious evening gown of deep crimson velvet, her hair styled into soft waves cascading over one shoulder. She wears diamond earrings and a matching necklace. The backdrop is a dimly lit grand ballroom, with golden chandeliers casting a warm glow. Her expression is one of quiet sophistication.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "938be427eeb04a37bb699f48786787c3",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 4/10: Close up photo of ext woman with her hair tied into a tight, elegant bun, wearing a high-tech metallic silver suit with glowing blue lines integrated into the design. The futuristic backdrop features holographic screens and neon lights. Her intense expression suggests intelligence and focus.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "e56d500dad614136bd8c773a10081acf",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 5/10: Medium shot photo of ext woman wearing a traditional sari in vibrant shades of gold and red, with intricate embroidery. She has a delicate bindi on her forehead, and her jewelry includes gold jhumkas and a necklace with emerald accents. The backdrop is a palace courtyard at sunset, with warm orange hues highlighting her serene expression.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "4127cb5cac35459693e4e3f89baeb818",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 6/10: Medium shot photo of ext woman dressed in a rugged explorer outfit: a khaki shirt with rolled-up sleeves, a leather hat, and a bandana around her neck. Her expression is determined, with a hint of adventure. The backdrop is a dense jungle, with sunlight streaming through the leaves.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "b941b7e6426f4700b3ee20290b18b36d",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 7/10: Medium shot photo of ext woman in a glowing elven robe of silver and white, her ears subtly pointed, and a delicate circlet of intertwined gold leaves adorning her head. The magical forest backdrop features luminous plants and a soft mist illuminated by moonlight. Her expression is wise and ethereal.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "fe96f984b8a54bf28233199afabe2bca",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 8/10: Full shot photo of ext woman wearing an elaborate black and gold masquerade mask, her gown shimmering with sequins that catch the light. Her lips are painted a deep red, and her eyes peer through the mask with a mysterious allure. The backdrop is a Venetian-style ballroom filled with masked figures.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "fda0102c3ca24c4a8efdad2e37c7d31d",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 9/10: Full shot photo of ext woman in a tailored white blazer over a silk blouse, her hair in a polished low bun. Her accessories include minimalist gold hoop earrings and a smartwatch. The backdrop is a modern corporate office with floor-to-ceiling windows overlooking a bustling cityscape.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "eaddf4e7efc24fa3971fbd9e4595cf94",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 10/10: Full shot photo of ext woman with flowing red hair adorned with wildflowers, wearing a simple yet elegant white linen dress. Her expression is serene, and her freckles are highlighted by the golden hour light. The backdrop is a lush meadow with tall grass swaying in the gentle breeze.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "09476e1b429548e49ef625f935fe3b92",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "prompts = [\n",
    "    \"Close up photo of ext woman wearing an exquisite white lace dress with intricate floral patterns, a delicate silver necklace, and pearl earrings. Her makeup is minimal, with a natural glow accentuating her features. The studio backdrop is a soft beige gradient, with professional lighting creating gentle shadows and highlights. Her expression is calm and poised, exuding timeless elegance.\",\n",
    "    \"Close up photo of ext woman wearing a sleek black leather jacket over a white turtleneck sweater. Her hair is styled in a modern bob, and she has a piercing gaze that conveys confidence. The backdrop is an urban skyline at dusk, with golden light reflecting off glass buildings. Subtle HDR lighting enhances the depth of her features.\",\n",
    "    \"Close up photo of ext woman in a luxurious evening gown of deep crimson velvet, her hair styled into soft waves cascading over one shoulder. She wears diamond earrings and a matching necklace. The backdrop is a dimly lit grand ballroom, with golden chandeliers casting a warm glow. Her expression is one of quiet sophistication.\",\n",
    "    \"Close up photo of ext woman with her hair tied into a tight, elegant bun, wearing a high-tech metallic silver suit with glowing blue lines integrated into the design. The futuristic backdrop features holographic screens and neon lights. Her intense expression suggests intelligence and focus.\",\n",
    "    \"Medium shot photo of ext woman wearing a traditional sari in vibrant shades of gold and red, with intricate embroidery. She has a delicate bindi on her forehead, and her jewelry includes gold jhumkas and a necklace with emerald accents. The backdrop is a palace courtyard at sunset, with warm orange hues highlighting her serene expression.\",\n",
    "    \"Medium shot photo of ext woman dressed in a rugged explorer outfit: a khaki shirt with rolled-up sleeves, a leather hat, and a bandana around her neck. Her expression is determined, with a hint of adventure. The backdrop is a dense jungle, with sunlight streaming through the leaves.\",\n",
    "    \"Medium shot photo of ext woman in a glowing elven robe of silver and white, her ears subtly pointed, and a delicate circlet of intertwined gold leaves adorning her head. The magical forest backdrop features luminous plants and a soft mist illuminated by moonlight. Her expression is wise and ethereal.\",\n",
    "    \"Full shot photo of ext woman wearing an elaborate black and gold masquerade mask, her gown shimmering with sequins that catch the light. Her lips are painted a deep red, and her eyes peer through the mask with a mysterious allure. The backdrop is a Venetian-style ballroom filled with masked figures.\",\n",
    "    \"Full shot photo of ext woman in a tailored white blazer over a silk blouse, her hair in a polished low bun. Her accessories include minimalist gold hoop earrings and a smartwatch. The backdrop is a modern corporate office with floor-to-ceiling windows overlooking a bustling cityscape.\",\n",
    "    \"Full shot photo of ext woman with flowing red hair adorned with wildflowers, wearing a simple yet elegant white linen dress. Her expression is serene, and her freckles are highlighted by the golden hour light. The backdrop is a lush meadow with tall grass swaying in the gentle breeze.\"\n",
    "]\n",
    "\n",
    "save_dir = \"taylor_lora_1500steps_generated_images\"\n",
    "for i, prompt in enumerate(prompts):\n",
    "    print(f\"Generating images for prompt {i+1}/{len(prompts)}: {prompt}\")\n",
    "    images = flux_dev_pipe(prompt, \n",
    "                  num_inference_steps=35, \n",
    "                  num_images_per_prompt=1, \n",
    "                  width=1024,\n",
    "                  height=1024,\n",
    "                  guidance_scale=3.5,\n",
    "                  generator=torch.Generator(\"cpu\").manual_seed(1024)).images\n",
    "\n",
    "    if not os.path.exists(save_dir):\n",
    "        os.makedirs(save_dir)\n",
    "               \n",
    "    images[0].save(f\"{save_dir}/prompt_{i+1}.jpg\")"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "8b710166-d3f0-49fa-9dbc-231d455ae949",
   "metadata": {},
   "source": [
    "### Final steps"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "id": "1284085a-c2f6-4e3e-9add-8b086b7f898b",
   "metadata": {},
   "outputs": [],
   "source": [
    "flux_dev_pipe.fuse_lora(lora_scale=0)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "id": "7e4e1195-bdec-4dc9-b291-4b95b0703009",
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "FluxPipeline {\n",
       "  \"_class_name\": \"FluxPipeline\",\n",
       "  \"_diffusers_version\": \"0.32.1\",\n",
       "  \"_name_or_path\": \"black-forest-labs/FLUX.1-dev\",\n",
       "  \"feature_extractor\": [\n",
       "    null,\n",
       "    null\n",
       "  ],\n",
       "  \"image_encoder\": [\n",
       "    null,\n",
       "    null\n",
       "  ],\n",
       "  \"scheduler\": [\n",
       "    \"diffusers\",\n",
       "    \"FlowMatchEulerDiscreteScheduler\"\n",
       "  ],\n",
       "  \"text_encoder\": [\n",
       "    \"transformers\",\n",
       "    \"CLIPTextModel\"\n",
       "  ],\n",
       "  \"text_encoder_2\": [\n",
       "    \"transformers\",\n",
       "    \"T5EncoderModel\"\n",
       "  ],\n",
       "  \"tokenizer\": [\n",
       "    \"transformers\",\n",
       "    \"CLIPTokenizer\"\n",
       "  ],\n",
       "  \"tokenizer_2\": [\n",
       "    \"transformers\",\n",
       "    \"T5TokenizerFast\"\n",
       "  ],\n",
       "  \"transformer\": [\n",
       "    \"diffusers\",\n",
       "    \"FluxTransformer2DModel\"\n",
       "  ],\n",
       "  \"vae\": [\n",
       "    \"diffusers\",\n",
       "    \"AutoencoderKL\"\n",
       "  ]\n",
       "}"
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "lora_path = \"/workspace/taylor-rank16-multi-1e-4-constant/pytorch_lora_weights.safetensors\"\n",
    "flux_dev_pipe.load_lora_weights(lora_path)\n",
    "flux_dev_pipe.to(\"cuda\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "id": "c0d25993-fd61-4dd3-947f-7255e691ac48",
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 1/10: Close up photo of ext woman wearing an exquisite white lace dress with intricate floral patterns, a delicate silver necklace, and pearl earrings. Her makeup is minimal, with a natural glow accentuating her features. The studio backdrop is a soft beige gradient, with professional lighting creating gentle shadows and highlights. Her expression is calm and poised, exuding timeless elegance.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "cb6a2eab3f2e408d9080cb333895c910",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 2/10: Close up photo of ext woman wearing a sleek black leather jacket over a white turtleneck sweater. Her hair is styled in a modern bob, and she has a piercing gaze that conveys confidence. The backdrop is an urban skyline at dusk, with golden light reflecting off glass buildings. Subtle HDR lighting enhances the depth of her features.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "a930af1ed23945e783081e0578dae6b5",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 3/10: Close up photo of ext woman in a luxurious evening gown of deep crimson velvet, her hair styled into soft waves cascading over one shoulder. She wears diamond earrings and a matching necklace. The backdrop is a dimly lit grand ballroom, with golden chandeliers casting a warm glow. Her expression is one of quiet sophistication.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "9ed94025a7fa449ab20b5fedf9e191a9",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 4/10: Close up photo of ext woman with her hair tied into a tight, elegant bun, wearing a high-tech metallic silver suit with glowing blue lines integrated into the design. The futuristic backdrop features holographic screens and neon lights. Her intense expression suggests intelligence and focus.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "ba126d975abd4765ae4360a414dc7111",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 5/10: Medium shot photo of ext woman wearing a traditional sari in vibrant shades of gold and red, with intricate embroidery. She has a delicate bindi on her forehead, and her jewelry includes gold jhumkas and a necklace with emerald accents. The backdrop is a palace courtyard at sunset, with warm orange hues highlighting her serene expression.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "ec8bcbb9cb1647ac96bda4d2ffdb9118",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 6/10: Medium shot photo of ext woman dressed in a rugged explorer outfit: a khaki shirt with rolled-up sleeves, a leather hat, and a bandana around her neck. Her expression is determined, with a hint of adventure. The backdrop is a dense jungle, with sunlight streaming through the leaves.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "4698da0d342942b69fd567cd009ce469",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 7/10: Medium shot photo of ext woman in a glowing elven robe of silver and white, her ears subtly pointed, and a delicate circlet of intertwined gold leaves adorning her head. The magical forest backdrop features luminous plants and a soft mist illuminated by moonlight. Her expression is wise and ethereal.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "433f7cd387da4342be4c27f06e220e6f",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 8/10: Full shot photo of ext woman wearing an elaborate black and gold masquerade mask, her gown shimmering with sequins that catch the light. Her lips are painted a deep red, and her eyes peer through the mask with a mysterious allure. The backdrop is a Venetian-style ballroom filled with masked figures.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "842b11a0b7c84722aa1f0842af3adc81",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 9/10: Full shot photo of ext woman in a tailored white blazer over a silk blouse, her hair in a polished low bun. Her accessories include minimalist gold hoop earrings and a smartwatch. The backdrop is a modern corporate office with floor-to-ceiling windows overlooking a bustling cityscape.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "19b0042870534f5ca1e73ac2869b7570",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating images for prompt 10/10: Full shot photo of ext woman with flowing red hair adorned with wildflowers, wearing a simple yet elegant white linen dress. Her expression is serene, and her freckles are highlighted by the golden hour light. The backdrop is a lush meadow with tall grass swaying in the gentle breeze.\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "7977fad788e04b5bb5c014aadc6cd468",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "  0%|          | 0/35 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "prompts = [\n",
    "    \"Close up photo of ext woman wearing an exquisite white lace dress with intricate floral patterns, a delicate silver necklace, and pearl earrings. Her makeup is minimal, with a natural glow accentuating her features. The studio backdrop is a soft beige gradient, with professional lighting creating gentle shadows and highlights. Her expression is calm and poised, exuding timeless elegance.\",\n",
    "    \"Close up photo of ext woman wearing a sleek black leather jacket over a white turtleneck sweater. Her hair is styled in a modern bob, and she has a piercing gaze that conveys confidence. The backdrop is an urban skyline at dusk, with golden light reflecting off glass buildings. Subtle HDR lighting enhances the depth of her features.\",\n",
    "    \"Close up photo of ext woman in a luxurious evening gown of deep crimson velvet, her hair styled into soft waves cascading over one shoulder. She wears diamond earrings and a matching necklace. The backdrop is a dimly lit grand ballroom, with golden chandeliers casting a warm glow. Her expression is one of quiet sophistication.\",\n",
    "    \"Close up photo of ext woman with her hair tied into a tight, elegant bun, wearing a high-tech metallic silver suit with glowing blue lines integrated into the design. The futuristic backdrop features holographic screens and neon lights. Her intense expression suggests intelligence and focus.\",\n",
    "    \"Medium shot photo of ext woman wearing a traditional sari in vibrant shades of gold and red, with intricate embroidery. She has a delicate bindi on her forehead, and her jewelry includes gold jhumkas and a necklace with emerald accents. The backdrop is a palace courtyard at sunset, with warm orange hues highlighting her serene expression.\",\n",
    "    \"Medium shot photo of ext woman dressed in a rugged explorer outfit: a khaki shirt with rolled-up sleeves, a leather hat, and a bandana around her neck. Her expression is determined, with a hint of adventure. The backdrop is a dense jungle, with sunlight streaming through the leaves.\",\n",
    "    \"Medium shot photo of ext woman in a glowing elven robe of silver and white, her ears subtly pointed, and a delicate circlet of intertwined gold leaves adorning her head. The magical forest backdrop features luminous plants and a soft mist illuminated by moonlight. Her expression is wise and ethereal.\",\n",
    "    \"Full shot photo of ext woman wearing an elaborate black and gold masquerade mask, her gown shimmering with sequins that catch the light. Her lips are painted a deep red, and her eyes peer through the mask with a mysterious allure. The backdrop is a Venetian-style ballroom filled with masked figures.\",\n",
    "    \"Full shot photo of ext woman in a tailored white blazer over a silk blouse, her hair in a polished low bun. Her accessories include minimalist gold hoop earrings and a smartwatch. The backdrop is a modern corporate office with floor-to-ceiling windows overlooking a bustling cityscape.\",\n",
    "    \"Full shot photo of ext woman with flowing red hair adorned with wildflowers, wearing a simple yet elegant white linen dress. Her expression is serene, and her freckles are highlighted by the golden hour light. The backdrop is a lush meadow with tall grass swaying in the gentle breeze.\"\n",
    "]\n",
    "\n",
    "negative_prompt = \"(old,wrinkles,deformed iris, deformed pupils, semi-realistic, cgi, 3d, render, sketch, cartoon, drawing, anime), text, cropped, out of frame, worst quality, low quality, jpeg artifacts, ugly, duplicate, morbid, mutilated, extra fingers, mutated hands, poorly drawn hands, poorly drawn face, mutation, deformed, blurry, dehydrated, bad anatomy, bad proportions, extra limbs, cloned face, disfigured, gross proportions, malformed limbs, missing arms, missing legs, extra arms, extra legs, fused fingers, too many fingers, long neck\"\n",
    "\n",
    "save_dir = \"taylor_lora_finalsteps_generated_images\"\n",
    "for i, prompt in enumerate(prompts):\n",
    "    print(f\"Generating images for prompt {i+1}/{len(prompts)}: {prompt}\")\n",
    "    images = flux_dev_pipe(prompt, \n",
    "                           negative_prompt=negative_prompt,\n",
    "                           num_inference_steps=35, \n",
    "                           num_images_per_prompt=1, \n",
    "                           width=1024,\n",
    "                           height=1024,\n",
    "                           guidance_scale=3.5,\n",
    "                           generator=torch.Generator(\"cpu\").manual_seed(1024)\n",
    "                          ).images\n",
    "\n",
    "    if not os.path.exists(save_dir):\n",
    "        os.makedirs(save_dir)\n",
    "               \n",
    "    images[0].save(f\"{save_dir}/prompt_{i+1}.jpg\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "4306f6c1-561b-43cc-a965-64f844dc9a89",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.12"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}