diff --git "a/Revised.ipynb" "b/Revised.ipynb" deleted file mode 100644--- "a/Revised.ipynb" +++ /dev/null @@ -1,35746 +0,0 @@ -{ - "cells": [ - { - "cell_type": "code", - "execution_count": null, - "metadata": { - "colab": { - "base_uri": "https://localhost:8080/" - }, - "id": "5xhZBPJobvEm", - "outputId": "df39269b-7c66-4b95-8d89-65bd2ef111aa" - }, - "outputs": [ - { - "output_type": "stream", - "name": "stdout", - "text": [ - "Collecting git+https://github.com/huggingface/diffusers.git\n", - " Cloning https://github.com/huggingface/diffusers.git to /tmp/pip-req-build-rfoppmb9\n", - " Running command git clone --filter=blob:none --quiet https://github.com/huggingface/diffusers.git /tmp/pip-req-build-rfoppmb9\n", - " Resolved https://github.com/huggingface/diffusers.git to commit fbca2e0a7a6643cd59de498ad7de0f63182edd19\n", - " Installing build dependencies ... \u001b[?25l\u001b[?25hdone\n", - " Getting requirements to build wheel ... \u001b[?25l\u001b[?25hdone\n", - " Preparing metadata (pyproject.toml) ... \u001b[?25l\u001b[?25hdone\n", - "Requirement already satisfied: importlib-metadata in /usr/local/lib/python3.10/dist-packages (from diffusers==0.21.0.dev0) (6.8.0)\n", - "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from diffusers==0.21.0.dev0) (3.12.2)\n", - "Collecting huggingface-hub>=0.13.2 (from diffusers==0.21.0.dev0)\n", - " Downloading huggingface_hub-0.16.4-py3-none-any.whl (268 kB)\n", - "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m268.8/268.8 kB\u001b[0m \u001b[31m5.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", - "\u001b[?25hRequirement already satisfied: numpy in /usr/local/lib/python3.10/dist-packages (from diffusers==0.21.0.dev0) (1.23.5)\n", - "Requirement already satisfied: regex!=2019.12.17 in /usr/local/lib/python3.10/dist-packages (from diffusers==0.21.0.dev0) (2023.6.3)\n", - "Requirement already satisfied: requests in /usr/local/lib/python3.10/dist-packages (from diffusers==0.21.0.dev0) (2.31.0)\n", - "Collecting safetensors>=0.3.1 (from diffusers==0.21.0.dev0)\n", - " Downloading safetensors-0.3.3-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[31m40.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", - "\u001b[?25hRequirement already satisfied: Pillow in /usr/local/lib/python3.10/dist-packages (from diffusers==0.21.0.dev0) (9.4.0)\n", - "Requirement already satisfied: fsspec in /usr/local/lib/python3.10/dist-packages (from huggingface-hub>=0.13.2->diffusers==0.21.0.dev0) (2023.6.0)\n", - "Requirement already satisfied: tqdm>=4.42.1 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub>=0.13.2->diffusers==0.21.0.dev0) (4.66.1)\n", - "Requirement already satisfied: pyyaml>=5.1 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub>=0.13.2->diffusers==0.21.0.dev0) (6.0.1)\n", - "Requirement already satisfied: typing-extensions>=3.7.4.3 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub>=0.13.2->diffusers==0.21.0.dev0) (4.7.1)\n", - "Requirement already satisfied: packaging>=20.9 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub>=0.13.2->diffusers==0.21.0.dev0) (23.1)\n", - "Requirement already satisfied: zipp>=0.5 in /usr/local/lib/python3.10/dist-packages (from importlib-metadata->diffusers==0.21.0.dev0) (3.16.2)\n", - "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests->diffusers==0.21.0.dev0) (3.2.0)\n", - "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests->diffusers==0.21.0.dev0) (3.4)\n", - "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests->diffusers==0.21.0.dev0) (2.0.4)\n", - "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests->diffusers==0.21.0.dev0) (2023.7.22)\n", - "Building wheels for collected packages: diffusers\n", - " Building wheel for diffusers (pyproject.toml) ... \u001b[?25l\u001b[?25hdone\n", - " Created wheel for diffusers: filename=diffusers-0.21.0.dev0-py3-none-any.whl size=1423421 sha256=2396313f9ac65a5f8a73eb20241bdc53fafa7a71317dd0ee5e541614db1dfe21\n", - " Stored in directory: /tmp/pip-ephem-wheel-cache-_valn2c1/wheels/4d/b7/a8/6f9549ceec5daad78675b857ac57d697c387062506520a7b50\n", - "Successfully built diffusers\n", - "Installing collected packages: safetensors, huggingface-hub, diffusers\n", - "Successfully installed diffusers-0.21.0.dev0 huggingface-hub-0.16.4 safetensors-0.3.3\n", - "Collecting git+https://github.com/huggingface/accelerate\n", - " Cloning https://github.com/huggingface/accelerate to /tmp/pip-req-build-ypjptnn3\n", - " Running command git clone --filter=blob:none --quiet https://github.com/huggingface/accelerate /tmp/pip-req-build-ypjptnn3\n", - " Resolved https://github.com/huggingface/accelerate to commit a16b843a1b8626d571e62f7301e3a478be2c04b4\n", - " Installing build dependencies ... \u001b[?25l\u001b[?25hdone\n", - " Getting requirements to build wheel ... \u001b[?25l\u001b[?25hdone\n", - " Preparing metadata (pyproject.toml) ... \u001b[?25l\u001b[?25hdone\n", - "Requirement already satisfied: numpy>=1.17 in /usr/local/lib/python3.10/dist-packages (from accelerate==0.23.0.dev0) (1.23.5)\n", - "Requirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.10/dist-packages (from accelerate==0.23.0.dev0) (23.1)\n", - "Requirement already satisfied: psutil in /usr/local/lib/python3.10/dist-packages (from accelerate==0.23.0.dev0) (5.9.5)\n", - "Requirement already satisfied: pyyaml in /usr/local/lib/python3.10/dist-packages (from accelerate==0.23.0.dev0) (6.0.1)\n", - "Requirement already satisfied: torch>=1.10.0 in /usr/local/lib/python3.10/dist-packages (from accelerate==0.23.0.dev0) (2.0.1+cu118)\n", - "Requirement already satisfied: huggingface-hub in /usr/local/lib/python3.10/dist-packages (from accelerate==0.23.0.dev0) (0.16.4)\n", - "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->accelerate==0.23.0.dev0) (3.12.2)\n", - "Requirement already satisfied: typing-extensions in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->accelerate==0.23.0.dev0) (4.7.1)\n", - "Requirement already satisfied: sympy in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->accelerate==0.23.0.dev0) (1.12)\n", - "Requirement already satisfied: networkx in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->accelerate==0.23.0.dev0) (3.1)\n", - "Requirement already satisfied: jinja2 in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->accelerate==0.23.0.dev0) (3.1.2)\n", - "Requirement already satisfied: triton==2.0.0 in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->accelerate==0.23.0.dev0) (2.0.0)\n", - "Requirement already satisfied: cmake in /usr/local/lib/python3.10/dist-packages (from triton==2.0.0->torch>=1.10.0->accelerate==0.23.0.dev0) (3.27.2)\n", - "Requirement already satisfied: lit in /usr/local/lib/python3.10/dist-packages (from triton==2.0.0->torch>=1.10.0->accelerate==0.23.0.dev0) (16.0.6)\n", - "Requirement already satisfied: fsspec in /usr/local/lib/python3.10/dist-packages (from huggingface-hub->accelerate==0.23.0.dev0) (2023.6.0)\n", - "Requirement already satisfied: requests in /usr/local/lib/python3.10/dist-packages (from huggingface-hub->accelerate==0.23.0.dev0) (2.31.0)\n", - "Requirement already satisfied: tqdm>=4.42.1 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub->accelerate==0.23.0.dev0) (4.66.1)\n", - "Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/dist-packages (from jinja2->torch>=1.10.0->accelerate==0.23.0.dev0) (2.1.3)\n", - "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests->huggingface-hub->accelerate==0.23.0.dev0) (3.2.0)\n", - "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests->huggingface-hub->accelerate==0.23.0.dev0) (3.4)\n", - "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests->huggingface-hub->accelerate==0.23.0.dev0) (2.0.4)\n", - "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests->huggingface-hub->accelerate==0.23.0.dev0) (2023.7.22)\n", - "Requirement already satisfied: mpmath>=0.19 in /usr/local/lib/python3.10/dist-packages (from sympy->torch>=1.10.0->accelerate==0.23.0.dev0) (1.3.0)\n", - "Building wheels for collected packages: accelerate\n", - " Building wheel for accelerate (pyproject.toml) ... \u001b[?25l\u001b[?25hdone\n", - " Created wheel for accelerate: filename=accelerate-0.23.0.dev0-py3-none-any.whl size=255798 sha256=ec590430d813a8ab800801daf4b6ec28251b9f15a075c31abf9188f71496f5a0\n", - " Stored in directory: /tmp/pip-ephem-wheel-cache-g1gxjke7/wheels/f6/c7/9d/1b8a5ca8353d9307733bc719107acb67acdc95063bba749f26\n", - "Successfully built accelerate\n", - "Installing collected packages: accelerate\n", - "Successfully installed accelerate-0.23.0.dev0\n", - "Collecting transformers\n", - " Downloading transformers-4.32.1-py3-none-any.whl (7.5 MB)\n", - "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m7.5/7.5 MB\u001b[0m \u001b[31m16.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", - "\u001b[?25hRequirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from transformers) (3.12.2)\n", - "Requirement already satisfied: huggingface-hub<1.0,>=0.15.1 in /usr/local/lib/python3.10/dist-packages (from transformers) (0.16.4)\n", - "Requirement already satisfied: numpy>=1.17 in /usr/local/lib/python3.10/dist-packages (from transformers) (1.23.5)\n", - "Requirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.10/dist-packages (from transformers) (23.1)\n", - "Requirement already satisfied: pyyaml>=5.1 in /usr/local/lib/python3.10/dist-packages (from transformers) (6.0.1)\n", - "Requirement already satisfied: regex!=2019.12.17 in /usr/local/lib/python3.10/dist-packages (from transformers) (2023.6.3)\n", - "Requirement already satisfied: requests in /usr/local/lib/python3.10/dist-packages (from transformers) (2.31.0)\n", - "Collecting tokenizers!=0.11.3,<0.14,>=0.11.1 (from transformers)\n", - " Downloading tokenizers-0.13.3-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (7.8 MB)\n", - "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m7.8/7.8 MB\u001b[0m \u001b[31m44.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", - "\u001b[?25hRequirement already satisfied: safetensors>=0.3.1 in /usr/local/lib/python3.10/dist-packages (from transformers) (0.3.3)\n", - "Requirement already satisfied: tqdm>=4.27 in /usr/local/lib/python3.10/dist-packages (from transformers) (4.66.1)\n", - "Requirement already satisfied: fsspec in /usr/local/lib/python3.10/dist-packages (from huggingface-hub<1.0,>=0.15.1->transformers) (2023.6.0)\n", - "Requirement already satisfied: typing-extensions>=3.7.4.3 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub<1.0,>=0.15.1->transformers) (4.7.1)\n", - "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests->transformers) (3.2.0)\n", - "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests->transformers) (3.4)\n", - "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests->transformers) (2.0.4)\n", - "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests->transformers) (2023.7.22)\n", - "Installing collected packages: tokenizers, transformers\n", - "Successfully installed tokenizers-0.13.3 transformers-4.32.1\n" - ] - } - ], - "source": [ - "!pip install git+https://github.com/huggingface/diffusers.git\n", - "!pip install git+https://github.com/huggingface/accelerate\n", - "!pip install --upgrade transformers" - ] - }, - { - "cell_type": "code", - "execution_count": null, - "metadata": { - "colab": { - "base_uri": "https://localhost:8080/" - }, - "id": "KuhLUa51fQfE", - "outputId": "68c7c483-d2ca-46b9-ea8d-e1b9dbaa4f0b" - }, - "outputs": [ - { - "output_type": "stream", - "name": "stdout", - "text": [ - "Collecting datasets\n", - " Downloading datasets-2.14.4-py3-none-any.whl (519 kB)\n", - "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m519.3/519.3 kB\u001b[0m \u001b[31m6.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", - "\u001b[?25hRequirement already satisfied: numpy>=1.17 in /usr/local/lib/python3.10/dist-packages (from datasets) (1.23.5)\n", - "Requirement already satisfied: pyarrow>=8.0.0 in /usr/local/lib/python3.10/dist-packages (from datasets) (9.0.0)\n", - "Collecting dill<0.3.8,>=0.3.0 (from datasets)\n", - " Downloading dill-0.3.7-py3-none-any.whl (115 kB)\n", - "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m115.3/115.3 kB\u001b[0m \u001b[31m6.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", - "\u001b[?25hRequirement already satisfied: pandas in /usr/local/lib/python3.10/dist-packages (from datasets) (1.5.3)\n", - "Requirement already satisfied: requests>=2.19.0 in /usr/local/lib/python3.10/dist-packages (from datasets) (2.31.0)\n", - "Requirement already satisfied: tqdm>=4.62.1 in /usr/local/lib/python3.10/dist-packages (from datasets) (4.66.1)\n", - "Collecting xxhash (from datasets)\n", - " Downloading xxhash-3.3.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (194 kB)\n", - "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m194.1/194.1 kB\u001b[0m \u001b[31m9.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", - "\u001b[?25hCollecting multiprocess (from datasets)\n", - " Downloading multiprocess-0.70.15-py310-none-any.whl (134 kB)\n", - "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m134.8/134.8 kB\u001b[0m \u001b[31m9.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", - "\u001b[?25hRequirement already satisfied: fsspec[http]>=2021.11.1 in /usr/local/lib/python3.10/dist-packages (from datasets) (2023.6.0)\n", - "Requirement already satisfied: aiohttp in /usr/local/lib/python3.10/dist-packages (from datasets) (3.8.5)\n", - "Requirement already satisfied: huggingface-hub<1.0.0,>=0.14.0 in /usr/local/lib/python3.10/dist-packages (from datasets) (0.16.4)\n", - "Requirement already satisfied: packaging in /usr/local/lib/python3.10/dist-packages (from datasets) (23.1)\n", - "Requirement already satisfied: pyyaml>=5.1 in /usr/local/lib/python3.10/dist-packages (from datasets) (6.0.1)\n", - "Requirement already satisfied: attrs>=17.3.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (23.1.0)\n", - "Requirement already satisfied: charset-normalizer<4.0,>=2.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (3.2.0)\n", - "Requirement already satisfied: multidict<7.0,>=4.5 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (6.0.4)\n", - "Requirement already satisfied: async-timeout<5.0,>=4.0.0a3 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (4.0.3)\n", - "Requirement already satisfied: yarl<2.0,>=1.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (1.9.2)\n", - "Requirement already satisfied: frozenlist>=1.1.1 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (1.4.0)\n", - "Requirement already satisfied: aiosignal>=1.1.2 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (1.3.1)\n", - "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from huggingface-hub<1.0.0,>=0.14.0->datasets) (3.12.2)\n", - "Requirement already satisfied: typing-extensions>=3.7.4.3 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub<1.0.0,>=0.14.0->datasets) (4.7.1)\n", - "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests>=2.19.0->datasets) (3.4)\n", - "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests>=2.19.0->datasets) (2.0.4)\n", - "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests>=2.19.0->datasets) (2023.7.22)\n", - "Requirement already satisfied: python-dateutil>=2.8.1 in /usr/local/lib/python3.10/dist-packages (from pandas->datasets) (2.8.2)\n", - "Requirement already satisfied: pytz>=2020.1 in /usr/local/lib/python3.10/dist-packages (from pandas->datasets) (2023.3)\n", - "Requirement already satisfied: six>=1.5 in /usr/local/lib/python3.10/dist-packages (from python-dateutil>=2.8.1->pandas->datasets) (1.16.0)\n", - "Installing collected packages: xxhash, dill, multiprocess, datasets\n", - "Successfully installed datasets-2.14.4 dill-0.3.7 multiprocess-0.70.15 xxhash-3.3.0\n", - "Requirement already satisfied: torchvision in /usr/local/lib/python3.10/dist-packages (0.15.2+cu118)\n", - "Requirement already satisfied: numpy in /usr/local/lib/python3.10/dist-packages (from torchvision) (1.23.5)\n", - "Requirement already satisfied: requests in /usr/local/lib/python3.10/dist-packages (from torchvision) (2.31.0)\n", - "Requirement already satisfied: torch==2.0.1 in /usr/local/lib/python3.10/dist-packages (from torchvision) (2.0.1+cu118)\n", - "Requirement already satisfied: pillow!=8.3.*,>=5.3.0 in /usr/local/lib/python3.10/dist-packages (from torchvision) (9.4.0)\n", - "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from torch==2.0.1->torchvision) (3.12.2)\n", - "Requirement already satisfied: typing-extensions in /usr/local/lib/python3.10/dist-packages (from torch==2.0.1->torchvision) (4.7.1)\n", - "Requirement already satisfied: sympy in /usr/local/lib/python3.10/dist-packages (from torch==2.0.1->torchvision) (1.12)\n", - "Requirement already satisfied: networkx in /usr/local/lib/python3.10/dist-packages (from torch==2.0.1->torchvision) (3.1)\n", - "Requirement already satisfied: jinja2 in /usr/local/lib/python3.10/dist-packages (from torch==2.0.1->torchvision) (3.1.2)\n", - "Requirement already satisfied: triton==2.0.0 in /usr/local/lib/python3.10/dist-packages (from torch==2.0.1->torchvision) (2.0.0)\n", - "Requirement already satisfied: cmake in /usr/local/lib/python3.10/dist-packages (from triton==2.0.0->torch==2.0.1->torchvision) (3.27.2)\n", - "Requirement already satisfied: lit in /usr/local/lib/python3.10/dist-packages (from triton==2.0.0->torch==2.0.1->torchvision) (16.0.6)\n", - "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests->torchvision) (3.2.0)\n", - "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests->torchvision) (3.4)\n", - "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests->torchvision) (2.0.4)\n", - "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests->torchvision) (2023.7.22)\n", - "Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/dist-packages (from jinja2->torch==2.0.1->torchvision) (2.1.3)\n", - "Requirement already satisfied: mpmath>=0.19 in /usr/local/lib/python3.10/dist-packages (from sympy->torch==2.0.1->torchvision) (1.3.0)\n", - "git-lfs is already the newest version (3.0.2-1ubuntu0.2).\n", - "0 upgraded, 0 newly installed, 0 to remove and 16 not upgraded.\n", - "Requirement already satisfied: tqdm in /usr/local/lib/python3.10/dist-packages (4.66.1)\n", - "Collecting bitsandbytes\n", - " Downloading bitsandbytes-0.41.1-py3-none-any.whl (92.6 MB)\n", - "\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m92.6/92.6 MB\u001b[0m \u001b[31m7.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", - "\u001b[?25hInstalling collected packages: bitsandbytes\n", - "Successfully installed bitsandbytes-0.41.1\n", - "Requirement already satisfied: torch in /usr/local/lib/python3.10/dist-packages (2.0.1+cu118)\n", - "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from torch) (3.12.2)\n", - "Requirement already satisfied: typing-extensions in /usr/local/lib/python3.10/dist-packages (from torch) (4.7.1)\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.1)\n", - "Requirement already satisfied: jinja2 in /usr/local/lib/python3.10/dist-packages (from torch) (3.1.2)\n", - "Requirement already satisfied: triton==2.0.0 in /usr/local/lib/python3.10/dist-packages (from torch) (2.0.0)\n", - "Requirement already satisfied: cmake in /usr/local/lib/python3.10/dist-packages (from triton==2.0.0->torch) (3.27.2)\n", - "Requirement already satisfied: lit in /usr/local/lib/python3.10/dist-packages (from triton==2.0.0->torch) (16.0.6)\n", - "Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/dist-packages (from jinja2->torch) (2.1.3)\n", - "Requirement already satisfied: mpmath>=0.19 in /usr/local/lib/python3.10/dist-packages (from sympy->torch) (1.3.0)\n", - "Requirement already satisfied: torchvision in /usr/local/lib/python3.10/dist-packages (0.15.2+cu118)\n", - "Requirement already satisfied: numpy in /usr/local/lib/python3.10/dist-packages (from torchvision) (1.23.5)\n", - "Requirement already satisfied: requests in /usr/local/lib/python3.10/dist-packages (from torchvision) (2.31.0)\n", - "Requirement already satisfied: torch==2.0.1 in /usr/local/lib/python3.10/dist-packages (from torchvision) (2.0.1+cu118)\n", - "Requirement already satisfied: pillow!=8.3.*,>=5.3.0 in /usr/local/lib/python3.10/dist-packages (from torchvision) (9.4.0)\n", - "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from torch==2.0.1->torchvision) (3.12.2)\n", - "Requirement already satisfied: typing-extensions in /usr/local/lib/python3.10/dist-packages (from torch==2.0.1->torchvision) (4.7.1)\n", - "Requirement already satisfied: sympy in /usr/local/lib/python3.10/dist-packages (from torch==2.0.1->torchvision) (1.12)\n", - "Requirement already satisfied: networkx in /usr/local/lib/python3.10/dist-packages (from torch==2.0.1->torchvision) (3.1)\n", - "Requirement already satisfied: jinja2 in /usr/local/lib/python3.10/dist-packages (from torch==2.0.1->torchvision) (3.1.2)\n", - "Requirement already satisfied: triton==2.0.0 in /usr/local/lib/python3.10/dist-packages (from torch==2.0.1->torchvision) (2.0.0)\n", - "Requirement already satisfied: cmake in /usr/local/lib/python3.10/dist-packages (from triton==2.0.0->torch==2.0.1->torchvision) (3.27.2)\n", - "Requirement already satisfied: lit in /usr/local/lib/python3.10/dist-packages (from triton==2.0.0->torch==2.0.1->torchvision) (16.0.6)\n", - "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests->torchvision) (3.2.0)\n", - "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests->torchvision) (3.4)\n", - "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests->torchvision) (2.0.4)\n", - "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests->torchvision) (2023.7.22)\n", - "Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/dist-packages (from jinja2->torch==2.0.1->torchvision) (2.1.3)\n", - "Requirement already satisfied: mpmath>=0.19 in /usr/local/lib/python3.10/dist-packages (from sympy->torch==2.0.1->torchvision) (1.3.0)\n" - ] - } - ], - "source": [ - "\n", - "!pip install datasets\n", - "\n", - "\n", - "!pip install torchvision\n", - "!sudo apt -qq install git-lfs\n", - "!git config --global credential.helper store\n", - "!pip install tqdm\n", - "!pip install bitsandbytes\n", - "!pip install torch\n", - "!pip install torchvision" - ] - }, - { - "cell_type": "code", - "execution_count": null, - "metadata": { - "id": "t6BleLJZgKR0" - }, - "outputs": [], - "source": [ - "from dataclasses import dataclass\n", - "from datasets import load_dataset\n", - "from torchvision import transforms\n", - "from accelerate.state import AcceleratorState\n", - "import math\n", - "import os\n", - "import numpy as np\n", - "import accelerate\n", - "from accelerate import Accelerator\n", - "from tqdm.auto import tqdm\n", - "from pathlib import Path\n", - "from accelerate import notebook_launcher\n", - "import torch.nn.functional as F\n", - "from diffusers.optimization import get_cosine_schedule_with_warmup\n", - "import torch\n", - "from PIL import Image\n", - "from diffusers import UNet2DModel\n", - "from transformers import CLIPTextModel, CLIPTokenizer\n", - "from diffusers import AutoencoderKL, DDPMScheduler, StableDiffusionPipeline, UNet2DConditionModel\n", - "from diffusers.optimization import get_scheduler\n", - "from huggingface_hub import create_repo, upload_folder, upload_file\n", - "import bitsandbytes as bnb\n", - "from transformers.utils import ContextManagers\n", - "from huggingface_hub import snapshot_download\n", - "\n", - "\n", - "@dataclass\n", - "class TrainingConfig:\n", - " pretrained_model_name_or_path = \"runwayml/stable-diffusion-v1-5\"\n", - " validation_prompts = [\"a dragon on a white background\",\" a fiery skull\", \"a skull\", \"a face\", \"a snake and skull\"]\n", - " image_size = 512 # the generated image resolution\n", - " train_batch_size = 2\n", - " eval_batch_size = 2 # how many images to sample during evaluation\n", - " num_epochs = 50\n", - " gradient_accumulation_steps = 1\n", - " lr_scheduler = \"constant\"\n", - " learning_rate = 1e-5\n", - " lr_warmup_steps = 500\n", - " save_image_epochs = 1\n", - " save_model_epochs = 1\n", - " token = \"xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\"\n", - " num_processes = 1\n", - " mixed_precision = \"fp16\" # `no` for float32, `fp16` for automatic mixed precision\n", - " output_dir = \"tattoo-diffusion\" # the model name locally and on the HF Hub\n", - "\n", - " push_to_hub = True # whether to upload the saved model to the HF Hub\n", - " hub_private_repo = False\n", - " overwrite_output_dir = True # overwrite the old model when re-running the notebook\n", - " seed = 0\n", - "\n", - "\n", - "config = TrainingConfig()" - ] - }, - { - "cell_type": "code", - "execution_count": null, - "metadata": { - "colab": { - "base_uri": "https://localhost:8080/", - "height": 1000, - "referenced_widgets": [ - "e7c8abcccd784964aca72de033b3a9b4", - "c4e510e9be1a4f9fa73389f8b14698bf", - "6a5295d7182c4a93b51ebd45f04977db", - "2bf7dfeba86d4f3e988a9bfd8476046b", - "e45786c671d24c9da480f79544ac4f7b", - "fc2452019d25434691a1bcc4c9a5c001", - "2fba1f16215c4acdb7388c0e46c69f00", - "e4f82f8cd16241d6ae273526d35755c1", - "bc732b9251824082a26698fe2f269e52", - "e803b4873bc0447bbd454c3a205acbf6", - "33c8cbb0567d4bdb9b5264cc1dc9d5be", - "6ac8d08b0dcb4f18bc863d7388cf74b8", - "421b3c67b5904b5fa6cd8118373a9f31", - "89fcd204863f4db3985e71d6214fbd74", - "83934ef75526407b8840a71552e566ff", - "09ac0cbc7c7d46f184a3fdb41b437590", - "d585924b92fd4559a94f98aca0e938ea", - "5cc7ce35e340475bbabe97d84e4f1a14", - "800137a3f97e4c8c9f218215fa3e9ec3", - "a26e2abdff00462ba9eb3d68aa343e71", - "5f17b06b51cf495a9af04778f2180492", - "bb9a8c9b0fb34f9e904e4a6c577f98b7", - "3d35e31a1ad649cfa92d7fe658595e92", - "0e318c825c7b434cb7d2b6a51fabe19c", - "f8295c24115b41e4b939a3415cf11be7", - "8e76401dc3e641e0a86e29a092f5fdd6", - "0cbb99a088d24594934ad934e680d0e4", - "e58d509aea3b4de08e50f71f2a974944", - "ec952c12a25e4b7aaf95a96ac72db879", - "7666d54d0a8c4e2f959f6f7019eb8262", - "b2f7bd6eee744a268501ebb908254f68", - "f4f5c04a5abf4162a05caa98a4160ab7", - "4c9108173e1f4505bb35f8fee2f31d9c", - "3de3866ec6b24b679ee7ef0da135163d", - "8c0954c5b0e04f08aded7cf86c5f9808", - "906a42b52c664e6ba69c8be1953785cc", - "d1855741284b4907a487035fded5700c", - "84db1025f5e8483881ec1949f699195f", - "543ac0f30679444c83bed986bab101ca", - "bae9259e5ca04885a1d6d0cf7bc8ac22", - "46b80dc41f604f818b90ef4caaab4082", - "cc7da46f4abd410aaca05150114800aa", - "8107892d05f5451e9d2b66e620a5e3ff", - "03512a5106ca4565a1879adb3f480386", - "adeff76046d24164abd8d67630ac2fb4", - "2f28daffd35b4fc39219b8a0daae9180", - "4907f618097a46ec9e833652ccbd6452", - "9c425727284d4cdd9cad729441878651", - "22250acd380c4f4ab68c1f6cdca5daef", - "c9f20dcac7ee49dd824c026e5cdacde2", - "f62f587dfa474dd2899064cab38c4841", - "d9f3959c5c4b4ef6b95dff581657bb1f", - "89112b7846ab430e90303b9d2c1504f1", - "4de11b57fe9c4fcf9b565bcbe549845e", - "d25f4243aff048708101d9a8451859a6", - "933cf703a9ba41a590263aa74af93375", - "a56e67d75863419eb71264a0120e5404", - "c293a95542f64fc6bea9773e90b04fb1", - "021a2c07acca4f149c1b23e6e6f113fd", - "96149f9bd0a04f43a81198eb86806e2b", - "2d1f1ba81b0749f7af0712a68259d756", - "6e361f020f6b4942ab3d66bfe9ae712d", - "0626e13119f34159af642409c0157285", - "ea45033f70f049c1bef74f7ca4d6e6d9", - "a26bab10aa3b444c868d1e041c8a0067", - "86b1e39d5ca346c4a4fb572420ba6bd5", - "e5070f49ddda498aa00dd785195efb2d", - "d152b1257db84b869521795694d72330", - "3fda4447d242477e8173ebb4fb0fec39", - "6ee217ea410844b298818e8c22fa668d", - "98b3fc82ce704d50a103505e0dfa6fce", - "b663ea8fc4564189a55e78c803d67092", - "ef915c8de87d49bfaea529d3502db2c2", - "0180977e691c42118d988ad5b166ddff", - "6b84545db6404f6f81142f59654b488b", - "175680570f2c413ba8397be54537c3ff", - "4ba320a87e2d4c73a46038a033f8450c", - "a46f1698f46341c58de1edbb81f2643b", - "034be6139e2c405b8864c3850f7ec0d7", - "7c29efcb2ab8409e9619dcb944471db6", - "a39a2ed6c7764c5c877e8c611fd08f71", - "25c6c0e692da4eb8bfd776b9e024e8a0", - "65e61cafe70c424c959b38aae947cdf2", - "bba7ba544e4f4213bbbba3571f7e29f0", - "853aff37200546f98e834fd80e8d2fcb", - "cffa39bcf8a143eaa859aff3c9d93981", - "c0733260ed134d908037f81861473ceb", - "f9b825895afd4b23a47c261a42baef7f", - "fa6a1fe9d494426d89f1cabd02922b2e", - "b4733052fd374186a015dcedda13d647", - "9743903da964449c8d0905de382cba84", - "facc8c1ba7b64ef395ff3541c5b5209c", - "7d332701834f4e52928490e5b47d6139", - "81062d099340477fa809e4fb0706987c", - "8a91ad227c3d4236a544beefdafb1d94", - "f8a986c53209467b8bd5c6427dc7d6bf", - "a3429831d1cf4155a0bc0263cc0c139c", - "e15dc5a119184e8b90be4c3aadd3d427", - "57f0e502c7ad40498732bb506c787b26", - "17ff063f85574934b6df44be57f1547f", - "0976150121b64c12b36bb5dd89240861", - "99838d5e320d48659258e478f8ba2795", - "56b1d9bd4fdf42d6bcbb5ad91046475b", - "06cddef900284c8ea286ce2f5c4d3a4b", - "313d876c0c214b3883c9142fdd7154b4", - "2e3265f20799465cb5c4719588a1fcaa", - "a47efc9548af43cf864c6a2c485cf825", - "e664410ef0ad4ff78d78dfe40e2f2de8", - "45fec58f6ad84773b8cecc8f1edef7cb", - "a2d88b9a32c74e1fb19254ff16eaf2f5", - "6377f8109b2e4f5fbb319c552fc6e79d", - "57fe017a148a4a3f9e761f837da29674", - "a8a13a1197044f1ca3be5ce4d926b623", - "2c8137055f964545be7983047e1cfd54", - "8d9aad3e0acb406e8fb8e3191e2f371f", - "123bea816f27488f9a4837d63f5fc79b", - "c924fd14a84044a4b2358af5988c82bf", - "b95103d848e348409a8ece4cbf8f2e7d", - "01f103dcb6da44a8b5eff7d3c736603f", - "c417697c283c4c87980201db01681add", - "3df799f6c96247a4b9da35bc44534d4f", - "1e35fb744b654c67a700420ee56adda5", - "d05f4bc2fea446fe83a611ca33638ffb", - "e25435d2696e44f29b485d993e717319", - "2b08f3aa0e644b888e08c8e8d6cd40f5", - "f5a1079897004ad99ad9f61521058fb2", - "d7a818dffb9642d1abab405227bf49e0", - "545bf88ac1fe4c15a437babab654e39e", - "2be5daa4cb9d44e59bb6c7eb78dc78ea", - "c2cc518cd09b4fb2a77fafbddedc3978", - "df34e71a4a4e47ea833c71beac06ea59", - "5928a9ed54de4ad09ec65f0b3aa519b0", - "1efd174777b741b69b8546ad803653cf", - "440610c997be41c79a8c6faab9384293", - "89bb922e98e345aca44999ffd5ec1694", - "6e905271c0fb4d1fb18b04a4642746ed", - "a8656076e5ba48668d21b5c2467d377b", - "b1de915ae74a4955bb1d2c54b3ab6a7b", - "11446da997024cb99670a0af8c19d9a3", - "dfbf12d8e7974e8482b01c38c40fb208", - "60065c2252394af2a785217cb209670a", - "ff29c5cc70884d1088007c777fe7e5c4", - "769b30831a7d4d34b311895949d77854", - "6daae8556ddc4fe2a66962ba610cd4a2", - "67e9aa0ff1d847b9b51f7fdff80db87b", - "79debb3170b648ed8ca0ac8ca5a88689", - "5ba27d008d9443439c59e33fe8f4d867", - "52a0d41ffbd44d7596165cacdfebc30e", - "97653530d0cb42e1b44067eac797a082", - "da886f9955bb4a18b279a3654685735c", - "e9ea8015efdf4a6980753a4240d05855", - "91f01e2f5c77403c9824b7d21b84aa41", - "481600c26afe47eab0a1f0e34994d431", - "c577c38989a64af18b1ec6a32c29590f", - "994ad52d3db24be59561b109302cdf81", - "d9f50a1228e849a5b990cfc2d14bb61d", - "e18df95b51d847a5bac2875f0a129730", - "f088f8f5fc7547f4bf803f2e07736d6c", - "0c6c8c45df514dc29de548c702398719", - "e2e55a9c5d404a5ab59f50308681f533", - "cf167d5a8df74e1c9f1a1c7b6905cd1f", - "533bf2004cbb409dbeca07dce29d44af", - "8277e851e10e4799b7232b62df8ed1db", - "fd565f83ceb1434599af837fbbeb965a", - "5e938def4921430983be46b35880da2c", - "e49663ee5c2440c689329d0cb7e6613b", - "15c0af47434145aebd5c751ba05655d8", - "fc065d3a6eba4b81b7c52e6fc1c27644", - "01b2bafc2d814248a2b44354d967f1d2", - "2cf640df683f4de9b987b97ee3baa2b4", - "e4f9b082b4824f378991386b49463517", - "e946923724384d5a84b53fde18d3b9e1", - "8e13978b7318460999081b3d5ff25be7", - "4ff97a9f65e14b3687ab1e33b0fb1db0", - "6840b7afc78e45088e1cd96bb346d0ce", - "201a7964475c40539131c6835c1fcbf1", - "4388c97c80db44e1a2f847436b5f6bfe", - "e0e40337b9e84a87ab36426d125148d9", - "ead47d74c0ce4452bd757cfe728cb5c1", - "1d87c21c83df41c6bd1612d3824035b4", - "7019172541e14d54b3f0d3f8e7ede04f", - "f299716e85aa47168d1c2f8239e1facd", - "b9b55bb117f94ccd9b632a2416b7e471", - "dae06a9a2ec54120a45fbcd8ffec669b", - "539b943b875a4d18ae9dc5c2ec281e41", - "174f5384d84d45ad89ba37fed826f29b", - "575220073f654fd4a2b7b849cd89fb4c", - "ba47f238d6a84e4c86406131e4a1daca", - "c6bfe144296546a19b804594c0286700", - "443ee1a95c6746d9b985fbbbde51f37e", - "59bcd3f8c5a84ad3ad6095e2633008f5", - "7b6972fcdf314aa5a654bb9c983c287c", - "404e28790fc44dd5a85d2acc72b61e81", - "68edcd4d03894d4da15986b61b3d79d7", - "66d4e923a594479d9ad7152c4334f193", - "18eda3552b294977b91edd163b4ac203", - "8a62b2f9f7c045ca83b8879d8e25c7f4", - "1064bca95edb41768915899823065568", - "7ea15b8aeb714c08988cb11f8e55fde7", - "1df2bfab5f164f34a2b679eb6c27420d", - "1c941a0317db4595a3390387f1268861", - "c3a8a0c9b0a24de1bbfbe51604741afd", - "1d8e2bf3d4a2462296abb97e88ab2632", - "c23cf262ae61495cb34398246526596c", - "ea7076f71d944cd0a0eedab1e7492712", - "b537c21aaa2a41378c67955278b25ed2", - "295d56bf2d664e5aab2c42a3e4ba5915", - "3788108d45314123aff8684cce221486", - "94ab6efa950f45209cbf7b8bd76736ea", - "4aac4041e09d41c3b5889816ccbf7bc0", - "81ce1725f15440fbaf1ecce850a6ed53", - "8a42b68204bf46c58076ca26279670cc", - "09528880137d4e89ad52ca45e800733d", - "ddbc40737ff7456a9825a400fff54420", - "8e7afd709e0c4026aea38e2e2b48e4a7", - "9b05de629c8d47bfab930df7ea5b7915", - "2a73e6ed7800468a8ec0037e55629d22", - "1e0984b6a0f141e48dfddb7d48ba14bd", - "731059f11e5d41ddb3ad61c8cf78b62c", - "1047f54a999b4c7e9e9669f6512bc6ff", - "eca96ef1ba6e4bdf8092a3c91034cf23", - "a5985a6160844ba2824dd8ef9d0aefbb", - "179f4631de644b48ad2bae3959c51520", - "98cd521d673a40888196797803ad38f1", - "d935018fd41442eea3176587aa2a8b31", - "1abd7a7cc5ef4dd69f33419d5bfe66bd", - "2424536dddf141ffb3592f3862b20d22", - "0d06a6cf018f45f9bf29a5ae04db6e8f", - "666fd1e2c646455d9a711a74e902d61f", - "19cff7b8d276492195f106ecea9ffcbb", - "a24b4c12472b46b296ad0c084858b5a2", - "ab7a183bf7c94356b2ac9689c471b2dd", - "462826190aa94fe6b8a2ec3e1a48353b", - "50b599d05676439d939f322a6b942a65", - "db20d91fc4ea49bb95d3e91e49a2b83e", - "4355c85dfbdc46738be7cf46081dc424", - "df1165b169ab40b1b9b2059790c26ea7", - "a490867ca1e44e60aaca1064c9c299f4", - "3040e02dab3b457a8b41bb128823cfd9", - "5248049eabb94c7abfdb69f349418702", - "15e231293f744805a3daa748fef7dddb", - "335b550210144bb4a00eb6bff35175af", - "1a675fbca4b54fde8c12e60a62a1fc5c", - "566194edec3b4ec097959df67b8b1ac7", - "241d2feeb39d47d4999951f00684b2f5", - "0c27047d1805453c9f4ce37f4bf8b581", - "ea872fcbdf1f42ab9cc375f1fdcdc89b", - "a29f2631909b4cf185400c2f921ec74a", - "669e317ba81d4f90b94e9dcc882702b0", - "a82374d53f894cf78a8a50126c83f8ba", - "ed3cfe43e18348899fa42bed6df9ceca", - "1bb3d1e636d14be1987573f5674dac43", - "2d8cbddf8a9c4991a5492c17225e5d7e", - "c70c5a40cc0444878a2a9b9b8f729a2f", - "be1cea9e6f1f4c21a770ed5d83fca527", - "50b22e95a1114725a1235703c071520b", - "0b63cd208cc54f01b63690792d192b16", - "4f92d8ddb873420b817fbe34e512af94", - "77a335ec5dcc4496aa78525cfbfe2a2c", - "ca1e6c6ff1ef4220b4d9f97cfeb99fe6", - "656992e6e0a841b187c755d050277d01", - "0095ce4e9195418a8cd052cf10d28a11", - "aee59a637b9a4a07a28fdba086113b5a", - "8e9f2fd4f51a45b894de3dd0c1713e3c", - "da6b9e4b9b8a455ca96f58b5db3bebab", - "40a8f1f3f6f7414ab5b75511c32a6404", - "f4dfa62ddb0b4c0c85ef891b0b0ea185", - "6d993e3407e84e85aaa5a3ae7c4ce655", - "8f23a4a5eb7943f69df4004c51373e23", - "2e010ea348d0409fac1dec453002fc7c", - "64bf0abd6b8e41e3bdca5bb0ae63d857", - "882fb079c74847c9b8239109fbacab36", - "a800e6c1c4494f6f83c9a950c552698a", - "d2dee2d3a7a44669bc3ad5bc73115044", - "d487b67c32e64fbd838fbb077ff9acac", - "fc41fa8b7bca4a479c5e29f8d1195b6a", - "e3611476ddd042348e3ed147ff17cc13", - "206db465f0c64165b20988dd61281acb", - "7ab4a67db346411b8694c7fb10a2d869", - "68f4431774db43c0bddd2a6cb9fa12f1", - "12b4a135b8bc4266b04b563d7f60fee4", - "07aba70432cf4a37a0686fcd82292cb8", - "a9343fab280e498b8bf6231b32a66670", - "34023da73538479398f45c668c8fd590", - "09f2d523f5c54a22ba9a8bea8c02baa4", - "d5647fe3a3ce45f19081f42478ede733", - "82a5d3903ec74cf4a201fc52321e072f", - "94d9e773e3de48e6b93523ec2a1764df", - "2ef3a6cd491f4980b419864f026bb705", - "c1148731eed14713a3b63f89332e799e", - "ae0ff348d8e940c798c52b94fa79abbc", - "2ac3891901c1437482a0273cfb3e2764", - "958bd9d900dc4741848b7bf0dad2c754", - "cf424f41a75949eda473456ae3ce2c07", - "b61861e1f61e4642a5ef748c693df4e8", - "06a5f0d0a0d34f6e8271f52d9ef95a51", - "63a85be552ed41afb06ab493e1934a0d", - "9a3b0ed9077c4e129ec0ddc331e039bb", - "f4699db0b6084f83967dfcf5f0d23ef7", - "243efc9d3bbf47229c64960652daa7b9", - "0c75311141b74810aa307c3e41acf5c1", - "2868eab4a91d4d0185ce090d7b076856", - "5150550905c849cbb097f6b7ac46c24d", - "60a6bb0b39004aeca949edab9e3c571e", - "ab4a95e66424457db8808ebd9f96b806", - "2656170825bd4c55bacbc04d97eaf095", - "b2c0b120a83b4c80a30adc6bb1aec797", - "b37867922bcc49c8a836f23084e3806d", - "8a7fee69884849a08439e4d7637d1371", - "7f6fb3d846694384974f2fcc1309ac05", - "58811cd413d24647971516efb9a53b5e", - "0fd23ce35e7e45bfb215131a976f9859", - "9e450d8342e040c6b1d8d80798f20f28", - "be75cdc2cf8a4533b6556f6c6cb93ce8", - "ae25ce65662249ffa4dc4b6578ff6b19", - "5eb437e5cdcb45e781423d7e10d0503e", - "a1d06d882af4424080cadbc513d10297", - "ec1f82e6e39c46e388d18fdbc58e4f77", - "9641c318f4634f90ae788412eb028acc", - "c1931a82a86540d59bf961acaa1f6938", - "8247aaf5d8424a6cbf5bdeed388d1870", - "88b89402d7d44815a75e0fa17ad46f33", - "7b1bd47113464c1193b8e61ee2da7058", - "139de2bf67f34beca38d6cfafc29ce12", - "449d7e044fec461aa0e863d1c82245d7", - "64a759008caa45ef84f9b194be8ccea5", - "0bf8350f2ce445d38252f7fed24a8972", - "a8efa93ba8ce46e18fb8da9966bae824", - "6de707662a4943eea5e91c0556deb7b8", - "36002b2bf5e04295a75b7301fc353e60", - "d6715915e8c944b8bec2f2a9617c1732", - "1598f46b95c04057933bea52aa96af29", - "8df75d6809584b7d92aa7cc4e807e015", - "7690abe4f45847288fde91c8c01b6189", - "f70a1673ac424a26bbb8215520ed1f30", - "0240240430f2422bad2b0fb206e0e4c7", - "31dd9fac369c40eda8b0814783cb2ee7", - "5318174db2804f0dad7846250643db3b", - "07c235d4f46746d19b69f415f473e297", - "f52ba89e5d7b470693335f6b9f987e1a", - "8d9bed457d444814bc0e467b819c3de5", - "12988f9de487461a942ff09c2f07c665", - "f2b0d29bdce4417abf89ee389e56ede0", - "64bdca8a9aa94f6bb454b0376aa30f93", - "7b657613a65b4ffab2d365ac263fbeae", - "21c4faa1b83945e097c1ad63ee5389ed", - "cd325b82241f47b4acb90c78a8e2f0af", - "e7ad147d76c842af86080658296c6da7", - "22ee03b2abc74d15959b13dd4519dc43", - "afbff1436618410784b168ac2293c8e7", - "bd14f3b4d51d4f77909c12e098d138e9", - "e483c9dff3bf43e1943647c1891e60d5", - "ee2f3d1d116c4f7ca7fdda4f47f0a00a", - "1ce5a73720954a99b8e067b903f48324", - "c889f93c3d4d445daadd807dcb2f7cd0", - "d76679eb567c4795a78aaef03cc50e59", - "59fb0287a8ad4b00b89d684e5e965ee1", - "b44a590705604a1482db915dba9d03f5", - "6d8928fdd797406e9f161b40dcb972e1", - "f966a342c5e74a3192fb44e8dc4f49ca", - "f56884dde93f4a58876e62fc8ef85e65", - "cdf0592f340e4bbd83c293f014523ad7", - "8971bc326aa04b4fa5d58f669954c649", - "145caf422db74826a24453da966d847e", - "fe7325f19e7b43c7bf8eed55719da686", - "25e32654d0fa41978b754325ee52ffdc", - "fa4bd7cc09234b13a7437a941d24d8af", - "6c76ee8a49f044bc92f32f5e735c8460", - "5103db3407b94213b74b649cdb07e7a0", - "7e0419dd045844dcb6ce4e2d1387b749", - "0d00ffab23c841d181a20672a3863765", - "b61c85afe1c34557badce92d78ab6c0d", - "3ed0cbdf5af6496a801c505e42c7d0be", - "ec6719f2f4cf4c91b1ba853768bfaf4e", - "78b9766ddd20481a9541373451a6c6e8", - "d9f91b5fea094365beb9a592d270d014", - "2a18f3000d494979b63bf733b6250214", - "9a0b5dc25a6b4d2280fc768a75e812e3", - "02af9d9b33b04848b2113f82d4dc67e1", - "1bf52889f34847158829672474c51b7c", - "7370e57a8fb4466e843f85dedbbcfc9b", - "e9b8c21f76f84a90bc6960d1e97b1469", - "54fdfd1d3f3a4431ab59713a3ab1181e", - "cf8dfb94f9b04db3a94654ab82404aa3", - "5e7db21f8c954d02a7f761bc1735200f", - "15bb981478d1489d9b59f2d755745678", - "c693391e71e1427c8bdc67927a5bf720", - "3a6470d84f57450a9fdfdef8249cc3e4", - "fc53dcdb92dc41f9a64c3b89856c4a2e", - "b57c92ac7db64bf9b9cd19e5beb7eaf6", - "d64c46967cb7442c911628be69d7da95", - "8649ab02986b4fea88a37e8aa247a086", - "0b7078bca6924274afe9436eb8e3faec", - "6a4e43cc8a954acd86ecd0ec5a24aa93", - "998533564541438a93e3dc54eb4b45cd", - "c36cd9760cd84a6e9be63b160e025201", - "4340183c19ef47cfa91522048c86e626", - "c83ec52c6f0949c7ab4faa7ce3146fc2", - "ca80aa2c9e1a4a8e8a8b84c7ed31ed64", - "e849e2da4dd24a0ebc407dc39344b390", - "81c978aa6b9b4cc4bbf67bfadf433371", - "5b85d735f76d4490880d2af7c5de6fb3", - "b04243cb21944c2986313a4b23744687", - "0ffaba02236c4058ba9a4413b5a8a24d", - "84b0bb4a7f3b4f51befd4f58bf13c320", - "e7ed41d56b104caa8eaf022e81ea68e6", - "8bcfcb10e34c4fdba93398feb9e29ccd", - "6b8e30c2b62b4eeb88fd644918e176d5", - "26bc6b6eef3348cdbe350b6ad7d91888", - "0d73738e4e5d4ed9affedc502fc1d71e", - "6614c18ec52745ebaebcdab381029945", - "ef99e486e4364f91812c64f507178672", - "35a65d0383aa43e580af241ecd2e3983", - "d52b9dff2ab24bd3b8a2573108b51b8d", - "a5efa8d5e3074f0da6bbc4c8c79a27cb", - "02218e321fa64cdc9ba370e7f2a1e417", - "260790391a42430582f8c72dca0bd395", - "be3e4da400824f05b4de458f6f4cb30b", - "2bfd2731832a4d51a270a9744cf604aa", - "6fceec74490c4527910a023c043ac30c", - "2627dca4f7fe4b668d2689a967990e19", - "06ce1b601ae2409bb7c40d43ddfb607d", - "895f6ef352c3488a872798668d3515d3", - "f56be5c6412c411bb0e6a1f1c9d8ee10", - "e565dda9bb0c4d3ead7e0172affe354b", - "83cbd0087033443b8a98b8b87c00df78", - "0370674c86414c1c9f8b7f8e6112f9e0", - "600fcf7b10164fa2a58f158b9037a724", - "5a8626fecab44e06a788b4e07823e66b", - "51d949a6f5a741ecaa76efc130aa56f3", - "b3e8ff82ead5431f891f3afbd24a9f1e", - "464479755ada459cba6169443cb04cae", - "691f17392e6649ed8f4f5d6616c934c9", - "75f2f1a151d049debaa6d14ea9882c84", - "8bf6f1e6cfdc4b1a8ae4cc8683fc01f6", - "40212d07fb334306ba6801edece13291", - "ffed0341c51a41d8ad661f9dd2ae0e0e", - "2769de565bab40ae81cf48765c4c0bf8", - "db31d26f85004fb1bf1619bda6b1b6e1", - "c24a15f2fed0482fbcb373f9f05b2ec2", - "cba645d8f74e40b69ab5fe2ac24e1869", - "243d16f381694a66b91979ec2de53878", - "9e1813f5084f4bbaa67277b2378eeabc", - "0e5865f59dca4457a5ef59b6f6253a3e", - "88cf8027e21d4c63b08171f2a4ab9e5d", - "e7f436c682cd4dd5b4a9fe03220f7cc4", - "79310d6c8803460f92409260baf584b4", - "11011104eab243b4b3743ba4847e375e", - "243d8a7926c7404fb553ca5c57a09e77", - "5ba33180bb784a30bfe78500dc359f7d", - "aecff9b33f234cacb37ad4480e6b218e", - "40fbe1e6f2134f6a97779cd814c55f65", - "8f25aa5d84e94d66b73ae18ef9a35938", - "75299024da8d4bbc93bace55aad7f57a", - "a2a3c3b3ec6d4428aaf15ac9ae0a81c4", - "d66e02f0906e42ac8823deb8ddd08fcb", - "0055f72836fd4d5086e8d218db7fd56e", - "db04da6bb3444603981131956f80ead4", - "c408489e12d947979a95d2114f3c9e5c", - "4de36148941243c4a5b766bc53d4de80", - "15807d5122a64bb9bf276411b472fbb5", - "af3f4817be764f368ab97f4bfa741bb9", - "be79363f95d74bcab8c8c4406e643480", - "7fdc5551967b4949b430845b85afab48", - "b03d5f6455ba4b1f819aa7eea0d5b663", - "86f5c73c93784517a533dc6b41990b87", - "c714084c014046fabccae3d9c0ad675b", - "8f66df7ae2d64c728ef2ed1ceb28ebcd", - "be403930d5d948ba82f980e27cc0f364", - "5144c67eca994f61baa52ac25b2f8350", - "b11900a5976e400c8d3cb419e959c101", - "7daf1af3f30843b18798b3f7a29aa9fc", - "161e6eedc74a484e871a93c743f7332e", - "828cdebc9c26414799d9e9e49ce5c4cb", - "eb393af90c8e4216973ab039de3a49c7", - "433f15679cf94e03988281cb4434135a", - "60cc687705f5437eae618bf19f536294", - "fe93c58f3d1b4c7485ac0f3ec619c30e", - "7d4ab5ab1f834579bd054fd2bf6309aa", - "7a47d3129eda495984186127b4454be1", - "ba1d311a4a80490eae3a275f82371e4d", - "b7a02eaf95c94cf499ae5e12b7d3cd63", - "a789aaff51c948b6800c5ca7f80ad214", - "f74a8a208955459cae314045f401eae2", - "eca53c8c52814eeeb75ab940d40dd79e", - "ddca0791ad1d43ae97ff17da49fb9287", - "a712147e0a4c4b3a8864792c0d3b69d2", - "a387d80dea8444d083e8b34dd96d0768", - "9b99779b0fb5428ab5b4adb085348db1", - "2dd21cad60984771bde7154c0c40f409", - "f93fb66e9a774abd8dd9511e52383a87", - "87e03d7493df4e37a245fdb34e30ef9a", - "7d63fa727bd44a5b84b905bd3da6ce61", - "6d9f1e79c96a475e81fc6677f8843e41", - "cdbac65dc0734dd9b78599fa13f118f3" - ] - }, - "id": "yBKWnM2p_qI6", - "outputId": "0a0a10ce-b190-4c22-ac31-1add7fe057df" - }, - "outputs": [ - { - "output_type": "display_data", - "data": { - "text/plain": [ - "Fetching 44 files: 0%| | 0/44 [00:00 0:\n", - " image_grid = make_grid(images, 1, len(config.validation_prompts))\n", - " image_grid.save(os.path.join(repo_folder, \"val_imgs_grid.png\"))\n", - " img_str += \"![val_imgs_grid](./val_imgs_grid.png)\\n\"\n", - " yaml = f\"\"\"\n", - "---\n", - "license: creativeml-openrail-m\n", - "base_model: {config.pretrained_model_name_or_path}\n", - "datasets:\n", - "- {config.dataset_name}\n", - "tags:\n", - "- stable-diffusion\n", - "- stable-diffusion-diffusers\n", - "- text-to-image\n", - "- diffusers\n", - "inference: true\n", - "---\n", - " \"\"\"\n", - " model_card = f\"\"\"\n", - "# Text-to-image finetuning - {repo_id}\n", - "\n", - "This pipeline was finetuned from **{config.pretrained_model_name_or_path}** on the **{config.dataset_name}** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: {config.validation_prompts}: \\n\n", - "{img_str}\n", - "\n", - "## Pipeline usage\n", - "\n", - "You can use the pipeline like so:\n", - "\n", - "```python\n", - "from diffusers import DiffusionPipeline\n", - "import torch\n", - "\n", - "pipeline = DiffusionPipeline.from_pretrained(\"{repo_id}\", torch_dtype=torch.float16)\n", - "prompt = \"{config.validation_prompts[0]}\"\n", - "image = pipeline(prompt).images[0]\n", - "image.save(\"my_image.png\")\n", - "```\n", - "\n", - "## Training info\n", - "\n", - "These are the key hyperparameters used during training:\n", - "\n", - "* Epochs: {config.num_epochs}\n", - "* Learning rate: {config.learning_rate}\n", - "* Batch size: {config.train_batch_size}\n", - "* Image resolution: {config.image_size}\n", - "* Mixed-precision: {config.mixed_precision}\n", - "\n", - "\"\"\"\n", - " with open(os.path.join(repo_folder, \"README.md\"), \"w\") as f:\n", - " f.write(yaml + model_card)\n", - "\n" - ] - }, - { - "cell_type": "code", - "execution_count": null, - "metadata": { - "id": "VbgnI0pJtsFQ" - }, - "outputs": [], - "source": [ - "def deepspeed_zero_init_disabled_context_manager():\n", - " \"\"\"\n", - " returns either a context list that includes one that will disable zero.Init or an empty context list\n", - " \"\"\"\n", - " deepspeed_plugin = AcceleratorState().deepspeed_plugin if accelerate.state.is_initialized() else None\n", - " if deepspeed_plugin is None:\n", - " return []\n", - "\n", - " return [deepspeed_plugin.zero3_init_context_manager(enable=False)]" - ] - }, - { - "cell_type": "code", - "execution_count": null, - "metadata": { - "id": "c6162g9pLz5r" - }, - "outputs": [], - "source": [ - "def train_loop(config, unet, vae, noise_scheduler, optimizer, train_dataloader, lr_scheduler):\n", - " repo_id = \"TejasNavada/tattoo-diffusion\"\n", - "\n", - " accelerator = Accelerator(\n", - " mixed_precision=config.mixed_precision,\n", - " gradient_accumulation_steps=config.gradient_accumulation_steps,\n", - " log_with=\"tensorboard\",\n", - " project_dir=os.path.join(config.output_dir, \"logs\"),\n", - " )\n", - " state_dict = lr_scheduler.state_dict()\n", - " print(state_dict)\n", - " if accelerator.is_main_process:\n", - " os.makedirs(config.output_dir,exist_ok=True)\n", - " accelerator.init_trackers(\"train_example\")\n", - "\n", - " unet, optimizer, train_dataloader, lr_scheduler = accelerator.prepare(\n", - " unet, optimizer, train_dataloader, lr_scheduler\n", - " )\n", - "\n", - "\n", - " text_encoder.to(accelerator.device, dtype=torch.float16)\n", - " vae.to(accelerator.device, dtype=torch.float16)\n", - " global_step = 0\n", - "\n", - " if(True):\n", - "\n", - " dirs = os.listdir(config.output_dir)\n", - " dirs = [d for d in dirs if d.startswith(\"checkpoint\")]\n", - " dirs = sorted(dirs, key=lambda x: int(x.split(\"-\")[1]))\n", - " path = dirs[-1] if len(dirs) > 0 else None\n", - " accelerator.print(f\"Resuming from checkpoint {path}\")\n", - " accelerator.load_state(os.path.join(config.output_dir, path))\n", - " global_step = int(path.split(\"-\")[1])\n", - "\n", - " start_epoch = global_step//len(train_dataloader)\n", - "\n", - " lr_scheduler.load_state_dict(state_dict)\n", - " print(lr_scheduler.get_last_lr())\n", - "\n", - " for epoch in range(start_epoch, config.num_epochs):\n", - " unet.train()\n", - "\n", - " progress_bar = tqdm(total=len(train_dataloader), disable=not accelerator.is_local_main_process)\n", - " progress_bar.set_description(f\"Epoch {epoch}\")\n", - "\n", - " for step, batch in enumerate(train_dataloader):\n", - "\n", - " # Convert images to latent space\n", - " latents = vae.encode(batch[\"pixel_values\"].to(torch.float16)).latent_dist.sample()\n", - " latents = latents * vae.config.scaling_factor\n", - "\n", - " # Sample noise that to add to the latents\n", - " noise = torch.randn_like(latents)\n", - "\n", - " bsz = latents.shape[0]\n", - "\n", - " # Sample a random timestep for each image\n", - " timesteps = torch.randint(\n", - " 0, noise_scheduler.config.num_train_timesteps, (bsz,), device=latents.device\n", - " ).long()\n", - " # Add noise to the latents according to the noise magnitude at each timestep\n", - " noisy_latents = noise_scheduler.add_noise(latents, noise, timesteps)\n", - " # Get the text embedding for conditioning\n", - " encoder_hidden_states = text_encoder(batch[\"input_ids\"])[0]\n", - " # Predict the noise residual and compute loss\n", - " with accelerator.accumulate(unet):\n", - "\n", - " model_pred = unet(noisy_latents,timesteps,encoder_hidden_states).sample\n", - "\n", - " loss = F.mse_loss(model_pred.float(),noise.float(), reduction=\"mean\")\n", - "\n", - " # Backpropagate\n", - " accelerator.backward(loss)\n", - " accelerator.clip_grad_norm_(unet.parameters(),1.0)\n", - "\n", - " optimizer.step()\n", - " lr_scheduler.step()\n", - " optimizer.zero_grad()\n", - "\n", - " progress_bar.update(1)\n", - " logs = {\"loss\": loss.detach().item(), \"lr\": lr_scheduler.get_last_lr()[0], \"step\": global_step}\n", - " progress_bar.set_postfix(**logs)\n", - " accelerator.log(logs, step=global_step)\n", - " global_step += 1\n", - "\n", - " if accelerator.is_main_process:\n", - "\n", - " if (epoch + 1) % config.save_image_epochs == 0 or epoch == config.num_epochs - 1:\n", - " images = evaluate(vae, text_encoder, tokenizer, unet, config, accelerator, epoch)\n", - " save_path = os.path.join(config.output_dir, f\"checkpoint-{global_step}\")\n", - " accelerator.save_state(save_path)\n", - " save_model_card(config, repo_id, images, repo_folder=config.output_dir)\n", - " upload_folder(\n", - " repo_id=repo_id,\n", - " folder_path=save_path,\n", - " path_in_repo=f\"checkpoint-{global_step}\",\n", - " commit_message=\"Latest Checkpoint\",\n", - " ignore_patterns=[\"step_*\", \"epoch_*\"],\n", - " )\n", - " upload_folder(\n", - " repo_id=repo_id,\n", - " folder_path=os.path.join(config.output_dir, \"samples\"),\n", - " path_in_repo=\"samples\",\n", - " commit_message=\"new samples\",\n", - " ignore_patterns=[\"step_*\", \"epoch_*\"],\n", - " )\n", - " upload_file(\n", - " path_or_fileobj=os.path.join(config.output_dir, \"README.md\"),\n", - " path_in_repo=\"README.md\",\n", - " repo_id=repo_id,\n", - " )\n", - "\n", - " unet = accelerator.unwrap_model(unet)\n", - " pipeline = StableDiffusionPipeline.from_pretrained(\n", - " config.pretrained_model_name_or_path,\n", - " text_encoder=text_encoder,\n", - " vae=vae,\n", - " unet=unet,\n", - " )\n", - " pipeline.save_pretrained(config.output_dir)\n", - " accelerator.end_training()\n", - "\n", - "\n", - "\n", - "\n", - "\n", - "\n", - "\n", - "\n", - "\n" - ] - }, - { - "cell_type": "code", - "execution_count": null, - "metadata": { - "colab": { - "base_uri": "https://localhost:8080/", - "height": 35 - }, - "id": "L21-Cx7NrghU", - "outputId": "e4f253ed-81e7-460a-ab48-a2ea3c27552c" - }, - "outputs": [ - { - "data": { - "application/vnd.google.colaboratory.intrinsic+json": { - "type": "string" - }, - "text/plain": [ - "'a dragon on a white background'" - ] - }, - "execution_count": 9, - "metadata": {}, - "output_type": "execute_result" - } - ], - "source": [ - "config.validation_prompts[0]" - ] - }, - { - "cell_type": "code", - "execution_count": null, - "metadata": { - "colab": { - "base_uri": "https://localhost:8080/" - }, - "id": "ofrTlboPpwX9", - "outputId": "6264d883-b773-4a07-dd02-685d8024ec97" - }, - "outputs": [ - { - "output_type": "stream", - "name": "stdout", - "text": [ - "Token is valid (permission: write).\n", - "Your token has been saved in your configured git credential helpers (store).\n", - "Your token has been saved to /root/.cache/huggingface/token\n", - "Login successful\n" - ] - } - ], - "source": [ - "from transformers.utils.hub import huggingface_hub\n", - "huggingface_hub.login(config.token, add_to_git_credential=True, new_session=True, write_permission=True)" - ] - }, - { - "cell_type": "code", - "execution_count": null, - "metadata": { - "id": "3o2O7BkjmNsB", - "colab": { - "base_uri": "https://localhost:8080/", - "height": 177, - "referenced_widgets": [ - "6da5cf65921d423fa7c2b0b714fbf8fd", - "b750bcdf5440429aad74a249a004d65d", - "f81f0ab145444636837ad04a791334e0", - "a214ae7c2d5c46348af2b303a2c1cd39", - "5e48d96ce2b5457099d3fd6e2526cacf", - "b5bc541e93a44c90a4b67774b4f52b5d", - "c86bcaf0c4a3410db88211cdfcb4b26a", - "6b2e25b496a94c57b74eb37a96e4df6a", - "bc53316be10145c5b4ebaccd6e570f43", - "64b06df8027442e48642b551bb0c8e7c", - "a240412dff8b48fa900f3f4df8115015", - "76da777f5efe466b92d0bac11deaa27b", - "24e7be2330c1417db1a24df6498afcc3", - "d6ab3763face4adc85dc0a04d839d113", - "4f112e9976d14572a023dce2d9ee88cf", - "ba649f1a145b4fcfbe3b7c8d46bbcb6b", - "331ee0587eb049aa9defbb5861e29777", - "fe60d19a79cf47d4b43edd42369f03f1", - "e1d5f563df114281b3fd3b8dd4226c49", - "a7663f24eabb4346aad95c3772355df7", - "db1764e8bbd74fbd827909a7aa4e3c55", - "9141cce75275431aa2acde37014be715", - "52004a24d08f4102819ca3b5128b5547", - "1da73aaa0ac34e0a9a95f5bc903a5d78", - "d13f28c25051434bb34206783ff8f85d", - "202ab9bbb4bb41e896c3bf417883781b", - "2b5790348e254d3fb1794401b250a706", - "fb2b219d14a64b0a98e1260f6b5babe4", - "600f2e9925cc4565aec51eb18581499c", - "6159fdc9c1dd4875943e5b3b951087a7", - "710c613a04b54d43bf60eda27140f7a1", - "1380ff6eafca42a8b08c3a5c73dc5da5", - "8bcd2b3e5aca454ea8f4ece2c81d872c", - "fd4012b31f054ec4b8764ccc472588ee", - "8fafd98431c64097a9490a68b4bee7ae", - "d0b1e5cb63374ac896820f343257d12e", - "01b15c4b1e934534adee140ef47cf73c", - "1b89b69116e44c17afd78a477b5f9247", - "e8d3f4df146f411397b78976c4373208", - "66e39789070248209749197c8776fbd3", - "0588f192ec3a4f88b0ac4283e4ed0e8d", - "69044d0de804447c85397aea7bab2666", - "9e49fd9bda9f44c882669f82bd81fc98", - "8f266444cfe544dbbb78a85a478f4f19", - "722f42b698f549dda1d871fbdb6716c0", - "972421cb3bb74e0baa3f1c39f67a170f", - "6feedf5bca504322a40aafb716d663c8", - "7ae8c8e6b1224b90a8cad70899ab0255", - "54355978b7d34abd92bf98cad37fd4e6", - "8f5de85c4c8b4e56a666667fec7cacfa", - "a0bc71177574450dbbc2c396943dd5e4", - "5f13341e0ddd4b9589b9dfa3f66df430", - "6713195e0ca04c288a5c0b6d5bab9acf", - "cff7357728b94e84b9ef24f783ef0ee4", - "1caef0df6843453b9190199edf5478a1" - ] - }, - "outputId": "f8f3a7b8-97d0-4592-e57d-844c52ecf9bd" - }, - "outputs": [ - { - "output_type": "display_data", - "data": { - "text/plain": [ - "Downloading (…)cheduler_config.json: 0%| | 0.00/308 [00:00 by passing `safety_checker=None`. Ensure that you abide to the conditions of the Stable Diffusion license and do not expose unfiltered results in services or applications open to the public. Both the diffusers team and Hugging Face strongly recommend to keep the safety filter enabled in all public facing circumstances, disabling it only for use-cases that involve analyzing network behavior or auditing its results. For more information, please have a look at https://github.com/huggingface/diffusers/pull/254 .\n" - ] - }, - { - "output_type": "display_data", - "data": { - "text/plain": [ - "pytorch_model.bin: 0%| | 0.00/3.44G [00:00 by passing `safety_checker=None`. Ensure that you abide to the conditions of the Stable Diffusion license and do not expose unfiltered results in services or applications open to the public. Both the diffusers team and Hugging Face strongly recommend to keep the safety filter enabled in all public facing circumstances, disabling it only for use-cases that involve analyzing network behavior or auditing its results. For more information, please have a look at https://github.com/huggingface/diffusers/pull/254 .\n" - ] - }, - { - "output_type": "display_data", - "data": { - "text/plain": [ - "pytorch_model.bin: 0%| | 0.00/3.44G [00:00 by passing `safety_checker=None`. Ensure that you abide to the conditions of the Stable Diffusion license and do not expose unfiltered results in services or applications open to the public. Both the diffusers team and Hugging Face strongly recommend to keep the safety filter enabled in all public facing circumstances, disabling it only for use-cases that involve analyzing network behavior or auditing its results. For more information, please have a look at https://github.com/huggingface/diffusers/pull/254 .\n" - ] - }, - { - "output_type": "display_data", - "data": { - "text/plain": [ - "random_states_0.pkl: 0%| | 0.00/14.7k [00:00 by passing `safety_checker=None`. Ensure that you abide to the conditions of the Stable Diffusion license and do not expose unfiltered results in services or applications open to the public. Both the diffusers team and Hugging Face strongly recommend to keep the safety filter enabled in all public facing circumstances, disabling it only for use-cases that involve analyzing network behavior or auditing its results. For more information, please have a look at https://github.com/huggingface/diffusers/pull/254 .\n" - ] - }, - { - "output_type": "display_data", - "data": { - "text/plain": [ - "optimizer.bin: 0%| | 0.00/1.73G [00:00