diff --git a/.idea/sagemaker-stablediffusion-quick-kit.iml b/.idea/sagemaker-stablediffusion-quick-kit.iml
new file mode 100644
index 0000000..d6ebd48
--- /dev/null
+++ b/.idea/sagemaker-stablediffusion-quick-kit.iml
@@ -0,0 +1,9 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="JAVA_MODULE" version="4">
+  <component name="NewModuleRootManager" inherit-compiler-output="true">
+    <exclude-output />
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>
\ No newline at end of file
diff --git a/fine-tuning/dreambooth/stablediffusion_xl_dreambooth_finetuning.zh.ipynb b/fine-tuning/dreambooth/stablediffusion_xl_dreambooth_finetuning.zh.ipynb
new file mode 100644
index 0000000..18b21c8
--- /dev/null
+++ b/fine-tuning/dreambooth/stablediffusion_xl_dreambooth_finetuning.zh.ipynb
@@ -0,0 +1,1274 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "a0e92bd0",
+   "metadata": {
+    "pycharm": {
+     "name": "#%% md\n"
+    }
+   },
+   "source": [
+    "### Dreambooth 模型微调\n",
+    "DreamBooth 是一种深度学习生成模型，用于微调现有的文本到图像模型，由 Google Research 和波士顿大学的研究人员于 2022 年开发。最初使用 Google 自己的 Imagen 文本到图像模型开发，DreamBooth 的实现可以应用到其他文本到图像模型，它可以让模型通过的三到五张图像对一个主题进行训练后生成更精细和个性化的输出。\n",
+    "\n",
+    "![](../../images/dreambooth.png)\n",
+    "\n",
+    "接下来我们将使用 DreamBooth 来微调我们的 stable diffusion xl模型.\n",
+    "\n",
+    "#### Notebook 步骤\n",
+    "1. 导入 boto3, sagemaker python SDK\n",
+    "2. 构建 dreambooth fine-tuning 镜像\n",
+    "3. 实现模型微调\n",
+    "   * 配置超参\n",
+    "   * 创建训练任务\n",
+    "4. 测试"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "eb9eb077",
+   "metadata": {
+    "pycharm": {
+     "name": "#%% md\n"
+    }
+   },
+   "source": [
+    "#### 1. 导入 boto3, sagemaker python SDK"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 52,
+   "id": "8314fc9b-c468-497b-abcc-259ec792154c",
+   "metadata": {
+    "pycharm": {
+     "name": "#%%\n"
+    },
+    "tags": []
+   },
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/home/ec2-user/anaconda3/envs/amazonei_pytorch_latest_p37/lib/python3.7/site-packages/boto3/compat.py:82: PythonDeprecationWarning: Boto3 will no longer support Python 3.7 starting December 13, 2023. To continue receiving service updates, bug fixes, and security updates please upgrade to Python 3.8 or later. More information can be found here: https://aws.amazon.com/blogs/developer/python-support-policy-updates-for-aws-sdks-and-tools/\n",
+      "  warnings.warn(warning, PythonDeprecationWarning)\n"
+     ]
+    }
+   ],
+   "source": [
+    "import sagemaker\n",
+    "import boto3\n",
+    "from sagemaker.pytorch import PyTorch\n",
+    "sagemaker_session = sagemaker.Session()\n",
+    "bucket = sagemaker_session.default_bucket()\n",
+    "role = sagemaker.get_execution_role()\n",
+    "account_id = boto3.client('sts').get_caller_identity().get('Account')\n",
+    "region_name = boto3.session.Session().region_name\n",
+    "\n",
+    "images_s3uri = 's3://{0}/dreambooth-xl/images/'.format(bucket)\n",
+    "models_s3uri = 's3://{0}/stable-diffusion/models/'.format(bucket)\n",
+    "dreambooth_s3uri = 's3://{0}/stable-diffusion/dreambooth/'.format(bucket)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "bd2a3178",
+   "metadata": {
+    "pycharm": {
+     "name": "#%% md\n"
+    }
+   },
+   "source": [
+    "#### 2. 构建 dreambooth xl fine-tuning 镜像"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "15c49cae-3336-4e34-aefd-c53e396f7b04",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "!cd sd_xl_dreambooth && git clone https://github.com/huggingface/diffusers"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "a7612e5a",
+   "metadata": {
+    "pycharm": {
+     "name": "#%%\n"
+    },
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "%%writefile Dockerfile\n",
+    "## You should change below region code to the region you used, here sample is use us-west-2\n",
+    "#From 763104351884.dkr.ecr.us-west-2.amazonaws.com/huggingface-pytorch-training:1.13.1-transformers4.26.0-gpu-py39-cu117-ubuntu20.04\n",
+    "From 763104351884.dkr.ecr.us-west-2.amazonaws.com/huggingface-pytorch-training:2.0.0-transformers4.28.1-gpu-py310-cu118-ubuntu20.04\n",
+    "\n",
+    "RUN pip install wandb\n",
+    "RUN pip install xformers==0.0.18\n",
+    "RUN pip install bitsandbytes\n",
+    "#RUN export TORCH_CUDA_ARCH_LIST=\"7.5 8.0 8.6\" && export FORCE_CUDA=\"1\" && pip install ninja triton==2.0.0.dev20221120 && git clone https://github.com/xieyongliang/xformers.git /tmp/xformers && cd /tmp/xformers && git submodule update --init --recursive && pip install -r requirements.txt && pip install -e . \n",
+    "\n",
+    "\n",
+    "ENV LANG=C.UTF-8\n",
+    "ENV PYTHONUNBUFFERED=TRUE\n",
+    "ENV PYTHONDONTWRITEBYTECODE=TRUE"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "9d70d510-caf7-4b48-95d4-f9bc2eaa0648",
+   "metadata": {},
+   "source": [
+    "* build & push docker镜像"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "f573e3c1-5e49-43cd-b71b-c858547192c9",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "## You should change below region code to the region you used, here sample is use us-west-2\n",
+    "!aws ecr get-login-password --region us-west-2 | docker login --username AWS --password-stdin 763104351884.dkr.ecr.us-west-2.amazonaws.com"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 53,
+   "id": "a69253dd-850f-41b7-b57a-437273648a46",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "## define repo name, should contain *sagemaker* in the name\n",
+    "repo_name = \"sd_xl_dreambooth_finetuning\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "20a370fa-bdf7-47a6-892d-f05adcf5904c",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "%%script env repo_name=$repo_name bash\n",
+    "\n",
+    "#!/usr/bin/env bash\n",
+    "\n",
+    "# This script shows how to build the Docker image and push it to ECR to be ready for use\n",
+    "# by SageMaker.\n",
+    "\n",
+    "# The argument to this script is the image name. This will be used as the image on the local\n",
+    "# machine and combined with the account and region to form the repository name for ECR.\n",
+    "# The name of our algorithm\n",
+    "algorithm_name=${repo_name}\n",
+    "\n",
+    "account=$(aws sts get-caller-identity --query Account --output text)\n",
+    "\n",
+    "# Get the region defined in the current configuration (default to us-west-2 if none defined)\n",
+    "region=$(aws configure get region)\n",
+    "region=${region:-us-west-2}\n",
+    "\n",
+    "fullname=\"${account}.dkr.ecr.${region}.amazonaws.com/${algorithm_name}:latest\"\n",
+    "\n",
+    "# If the repository doesn't exist in ECR, create it.\n",
+    "aws ecr describe-repositories --repository-names \"${algorithm_name}\" > /dev/null 2>&1\n",
+    "\n",
+    "if [ $? -ne 0 ]\n",
+    "then\n",
+    "    aws ecr create-repository --repository-name \"${algorithm_name}\" > /dev/null\n",
+    "fi\n",
+    "\n",
+    "# Get the login command from ECR and execute it directly\n",
+    "aws ecr get-login-password --region ${region}|docker login --username AWS --password-stdin ${fullname}\n",
+    "\n",
+    "# Build the docker image locally with the image name and then push it to ECR\n",
+    "# with the full name.\n",
+    "\n",
+    "docker build -t ${algorithm_name} .\n",
+    "docker tag ${algorithm_name} ${fullname}\n",
+    "\n",
+    "docker push ${fullname}"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "a8e01e5f-f53f-4443-a149-94d8c7126d8b",
+   "metadata": {
+    "tags": []
+   },
+   "source": [
+    "* 准备训练图像"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "285d6076-0746-4afa-ba61-0d53f91bd67a",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "from huggingface_hub import snapshot_download\n",
+    "\n",
+    "local_dir = \"./dog\"\n",
+    "snapshot_download(\n",
+    "    \"diffusers/dog-example\",\n",
+    "    local_dir=local_dir, repo_type=\"dataset\",\n",
+    "    ignore_patterns=\".gitattributes\",\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "c9f44953-ee8f-4b7a-bbfe-afad211b3224",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "!chmod -R 777 ./sd_xl_dreambooth\n",
+    "!./sd_xl_dreambooth/s5cmd sync ./dog/ $images_s3uri"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "1d843895",
+   "metadata": {
+    "pycharm": {
+     "name": "#%% md\n"
+    }
+   },
+   "source": [
+    "#### 3. 模型微调\n",
+    "\n",
+    "   * image_uri: ecr仓库中的 docker 镜像地址\n",
+    "   * instance_type: 用于训练任务的实例大小 , 建议使用 ml.g4dn.xlarge, ml.g5.xlarge\n",
+    "   * class_prompt: 提示词类别\n",
+    "   * instance_prompt: 用于你的图片的关键词\n",
+    "   * model_name: 预训练的模型名称\n",
+    "   "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 58,
+   "id": "720c66e8-8958-47f2-bfa7-c5252fde430e",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Overwriting ./sd_xl_dreambooth/train.sh\n"
+     ]
+    }
+   ],
+   "source": [
+    "%%writefile ./sd_xl_dreambooth/train.sh\n",
+    "\n",
+    "\n",
+    "mkdir -p /tmp/dog\n",
+    "ls -lt ./\n",
+    "chmod 777 ./s5cmd\n",
+    "\n",
+    "\n",
+    "cd diffusers && pip install -e .\n",
+    "cd examples/dreambooth/ && pip install -r requirements_sdxl.txt\n",
+    "\n",
+    "cp -r /opt/ml/input/data/images/* /tmp/dog/\n",
+    "\n",
+    "export MODEL_NAME=\"stabilityai/stable-diffusion-xl-base-1.0\"\n",
+    "export INSTANCE_DIR=\"/tmp/dog/\"\n",
+    "export OUTPUT_DIR=\"/tmp/ouput\"\n",
+    "#export OUTPUT_DIR=\"/opt/ml/model/\"\n",
+    "export VAE_PATH=\"madebyollin/sdxl-vae-fp16-fix\"\n",
+    "export dreambooth_s3uri=\"s3://sagemaker-us-west-2-687912291502/stable-diffusion/dreambooth/\"\n",
+    "\n",
+    "accelerate launch /opt/ml/code/diffusers/examples/dreambooth/train_dreambooth_lora_sdxl.py \\\n",
+    "  --gradient_checkpointing \\\n",
+    "  --use_8bit_adam \\\n",
+    "  --pretrained_model_name_or_path=$MODEL_NAME  \\\n",
+    "  --instance_data_dir=$INSTANCE_DIR \\\n",
+    "  --pretrained_vae_model_name_or_path=$VAE_PATH \\\n",
+    "  --output_dir=$OUTPUT_DIR \\\n",
+    "  --mixed_precision=\"fp16\" \\\n",
+    "  --instance_prompt=\"a photo of sks dog\" \\\n",
+    "  --resolution=1024 \\\n",
+    "  --train_batch_size=1 \\\n",
+    "  --gradient_accumulation_steps=4 \\\n",
+    "  --learning_rate=1e-5 \\\n",
+    "  --report_to=\"tensorboard\" \\\n",
+    "  --lr_scheduler=\"constant\" \\\n",
+    "  --lr_warmup_steps=0 \\\n",
+    "  --max_train_steps=500 \\\n",
+    "  --validation_prompt=\"A photo of sks dog in a bucket\" \\\n",
+    "  --validation_epochs=25 \\\n",
+    "  --seed=\"0\" \\\n",
+    "  --enable_xformers_memory_efficient_attention\n",
+    "\n",
+    "/opt/ml/code/s5cmd sync /tmp/ouput/ $dreambooth_s3uri/output/$(date +%Y-%m-%d-%H-%M-%S)/\n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "8c67611b-41a2-4977-9bfb-74342b2a5126",
+   "metadata": {},
+   "source": [
+    "* 本地跑测试"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 59,
+   "id": "641a23d6-0391-4c13-86e1-c81e226110c9",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "!pip list|grep -i xformers"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "264becae-cd2f-4b83-97c7-1e2469fa7f11",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "!./sd_xl_dreambooth/train.sh"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "9c569c81",
+   "metadata": {
+    "pycharm": {
+     "name": "#%% md\n"
+    }
+   },
+   "source": [
+    "   * 创建训练任务"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6535d22c-ab12-48c4-9989-5fabe4f31f69",
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Using provided s3_resource\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "INFO:sagemaker:Creating training-job with name: sd-xl-dreambooth-finetuning-high-2023-08-25-10-08-30-083\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "2023-08-25 10:08:34 Starting - Starting the training job...\n",
+      "2023-08-25 10:08:48 Starting - Preparing the instances for training......\n",
+      "2023-08-25 10:09:57 Downloading - Downloading input data...\n",
+      "2023-08-25 10:10:22 Training - Downloading the training image...........................\n",
+      "2023-08-25 10:14:48 Training - Training image download completed. Training in progress....\u001b[34mbash: cannot set terminal process group (-1): Inappropriate ioctl for device\u001b[0m\n",
+      "\u001b[34mbash: no job control in this shell\u001b[0m\n",
+      "\u001b[34m2023-08-25 10:15:27,344 sagemaker-training-toolkit INFO     Imported framework sagemaker_pytorch_container.training\u001b[0m\n",
+      "\u001b[34m2023-08-25 10:15:27,358 sagemaker-training-toolkit INFO     No Neurons detected (normal if no neurons installed)\u001b[0m\n",
+      "\u001b[34m2023-08-25 10:15:27,368 sagemaker_pytorch_container.training INFO     Block until all host DNS lookups succeed.\u001b[0m\n",
+      "\u001b[34m2023-08-25 10:15:27,369 sagemaker_pytorch_container.training INFO     Invoking user training script.\u001b[0m\n",
+      "\u001b[34m2023-08-25 10:15:29,659 sagemaker-training-toolkit INFO     No Neurons detected (normal if no neurons installed)\u001b[0m\n",
+      "\u001b[34m2023-08-25 10:15:29,683 sagemaker-training-toolkit INFO     No Neurons detected (normal if no neurons installed)\u001b[0m\n",
+      "\u001b[34m2023-08-25 10:15:29,706 sagemaker-training-toolkit INFO     No Neurons detected (normal if no neurons installed)\u001b[0m\n",
+      "\u001b[34m2023-08-25 10:15:29,715 sagemaker-training-toolkit INFO     Invoking user script\u001b[0m\n",
+      "\u001b[34mTraining Env:\u001b[0m\n",
+      "\u001b[34m{\n",
+      "    \"additional_framework_parameters\": {},\n",
+      "    \"channel_input_dirs\": {\n",
+      "        \"images\": \"/opt/ml/input/data/images\"\n",
+      "    },\n",
+      "    \"current_host\": \"algo-1\",\n",
+      "    \"current_instance_group\": \"homogeneousCluster\",\n",
+      "    \"current_instance_group_hosts\": [\n",
+      "        \"algo-1\"\n",
+      "    ],\n",
+      "    \"current_instance_type\": \"ml.g5.2xlarge\",\n",
+      "    \"distribution_hosts\": [],\n",
+      "    \"distribution_instance_groups\": [],\n",
+      "    \"framework_module\": \"sagemaker_pytorch_container.training:main\",\n",
+      "    \"hosts\": [\n",
+      "        \"algo-1\"\n",
+      "    ],\n",
+      "    \"hyperparameters\": {},\n",
+      "    \"input_config_dir\": \"/opt/ml/input/config\",\n",
+      "    \"input_data_config\": {\n",
+      "        \"images\": {\n",
+      "            \"TrainingInputMode\": \"File\",\n",
+      "            \"S3DistributionType\": \"FullyReplicated\",\n",
+      "            \"RecordWrapperType\": \"None\"\n",
+      "        }\n",
+      "    },\n",
+      "    \"input_dir\": \"/opt/ml/input\",\n",
+      "    \"instance_groups\": [\n",
+      "        \"homogeneousCluster\"\n",
+      "    ],\n",
+      "    \"instance_groups_dict\": {\n",
+      "        \"homogeneousCluster\": {\n",
+      "            \"instance_group_name\": \"homogeneousCluster\",\n",
+      "            \"instance_type\": \"ml.g5.2xlarge\",\n",
+      "            \"hosts\": [\n",
+      "                \"algo-1\"\n",
+      "            ]\n",
+      "        }\n",
+      "    },\n",
+      "    \"is_hetero\": false,\n",
+      "    \"is_master\": true,\n",
+      "    \"is_modelparallel_enabled\": null,\n",
+      "    \"is_smddpmprun_installed\": true,\n",
+      "    \"job_name\": \"sd-xl-dreambooth-finetuning-high-2023-08-25-10-08-30-083\",\n",
+      "    \"log_level\": 20,\n",
+      "    \"master_hostname\": \"algo-1\",\n",
+      "    \"model_dir\": \"/opt/ml/model\",\n",
+      "    \"module_dir\": \"s3://sagemaker-us-west-2-687912291502/sd-xl-dreambooth-finetuning-high-2023-08-25-10-08-30-083/source/sourcedir.tar.gz\",\n",
+      "    \"module_name\": \"train.sh\",\n",
+      "    \"network_interface_name\": \"eth0\",\n",
+      "    \"num_cpus\": 8,\n",
+      "    \"num_gpus\": 1,\n",
+      "    \"num_neurons\": 0,\n",
+      "    \"output_data_dir\": \"/opt/ml/output/data\",\n",
+      "    \"output_dir\": \"/opt/ml/output\",\n",
+      "    \"output_intermediate_dir\": \"/opt/ml/output/intermediate\",\n",
+      "    \"resource_config\": {\n",
+      "        \"current_host\": \"algo-1\",\n",
+      "        \"current_instance_type\": \"ml.g5.2xlarge\",\n",
+      "        \"current_group_name\": \"homogeneousCluster\",\n",
+      "        \"hosts\": [\n",
+      "            \"algo-1\"\n",
+      "        ],\n",
+      "        \"instance_groups\": [\n",
+      "            {\n",
+      "                \"instance_group_name\": \"homogeneousCluster\",\n",
+      "                \"instance_type\": \"ml.g5.2xlarge\",\n",
+      "                \"hosts\": [\n",
+      "                    \"algo-1\"\n",
+      "                ]\n",
+      "            }\n",
+      "        ],\n",
+      "        \"network_interface_name\": \"eth0\"\n",
+      "    },\n",
+      "    \"user_entry_point\": \"train.sh\"\u001b[0m\n",
+      "\u001b[34m}\u001b[0m\n",
+      "\u001b[34mEnvironment variables:\u001b[0m\n",
+      "\u001b[34mSM_HOSTS=[\"algo-1\"]\u001b[0m\n",
+      "\u001b[34mSM_NETWORK_INTERFACE_NAME=eth0\u001b[0m\n",
+      "\u001b[34mSM_HPS={}\u001b[0m\n",
+      "\u001b[34mSM_USER_ENTRY_POINT=train.sh\u001b[0m\n",
+      "\u001b[34mSM_FRAMEWORK_PARAMS={}\u001b[0m\n",
+      "\u001b[34mSM_RESOURCE_CONFIG={\"current_group_name\":\"homogeneousCluster\",\"current_host\":\"algo-1\",\"current_instance_type\":\"ml.g5.2xlarge\",\"hosts\":[\"algo-1\"],\"instance_groups\":[{\"hosts\":[\"algo-1\"],\"instance_group_name\":\"homogeneousCluster\",\"instance_type\":\"ml.g5.2xlarge\"}],\"network_interface_name\":\"eth0\"}\u001b[0m\n",
+      "\u001b[34mSM_INPUT_DATA_CONFIG={\"images\":{\"RecordWrapperType\":\"None\",\"S3DistributionType\":\"FullyReplicated\",\"TrainingInputMode\":\"File\"}}\u001b[0m\n",
+      "\u001b[34mSM_OUTPUT_DATA_DIR=/opt/ml/output/data\u001b[0m\n",
+      "\u001b[34mSM_CHANNELS=[\"images\"]\u001b[0m\n",
+      "\u001b[34mSM_CURRENT_HOST=algo-1\u001b[0m\n",
+      "\u001b[34mSM_CURRENT_INSTANCE_TYPE=ml.g5.2xlarge\u001b[0m\n",
+      "\u001b[34mSM_CURRENT_INSTANCE_GROUP=homogeneousCluster\u001b[0m\n",
+      "\u001b[34mSM_CURRENT_INSTANCE_GROUP_HOSTS=[\"algo-1\"]\u001b[0m\n",
+      "\u001b[34mSM_INSTANCE_GROUPS=[\"homogeneousCluster\"]\u001b[0m\n",
+      "\u001b[34mSM_INSTANCE_GROUPS_DICT={\"homogeneousCluster\":{\"hosts\":[\"algo-1\"],\"instance_group_name\":\"homogeneousCluster\",\"instance_type\":\"ml.g5.2xlarge\"}}\u001b[0m\n",
+      "\u001b[34mSM_DISTRIBUTION_INSTANCE_GROUPS=[]\u001b[0m\n",
+      "\u001b[34mSM_IS_HETERO=false\u001b[0m\n",
+      "\u001b[34mSM_MODULE_NAME=train.sh\u001b[0m\n",
+      "\u001b[34mSM_LOG_LEVEL=20\u001b[0m\n",
+      "\u001b[34mSM_FRAMEWORK_MODULE=sagemaker_pytorch_container.training:main\u001b[0m\n",
+      "\u001b[34mSM_INPUT_DIR=/opt/ml/input\u001b[0m\n",
+      "\u001b[34mSM_INPUT_CONFIG_DIR=/opt/ml/input/config\u001b[0m\n",
+      "\u001b[34mSM_OUTPUT_DIR=/opt/ml/output\u001b[0m\n",
+      "\u001b[34mSM_NUM_CPUS=8\u001b[0m\n",
+      "\u001b[34mSM_NUM_GPUS=1\u001b[0m\n",
+      "\u001b[34mSM_NUM_NEURONS=0\u001b[0m\n",
+      "\u001b[34mSM_MODEL_DIR=/opt/ml/model\u001b[0m\n",
+      "\u001b[34mSM_MODULE_DIR=s3://sagemaker-us-west-2-687912291502/sd-xl-dreambooth-finetuning-high-2023-08-25-10-08-30-083/source/sourcedir.tar.gz\u001b[0m\n",
+      "\u001b[34mSM_TRAINING_ENV={\"additional_framework_parameters\":{},\"channel_input_dirs\":{\"images\":\"/opt/ml/input/data/images\"},\"current_host\":\"algo-1\",\"current_instance_group\":\"homogeneousCluster\",\"current_instance_group_hosts\":[\"algo-1\"],\"current_instance_type\":\"ml.g5.2xlarge\",\"distribution_hosts\":[],\"distribution_instance_groups\":[],\"framework_module\":\"sagemaker_pytorch_container.training:main\",\"hosts\":[\"algo-1\"],\"hyperparameters\":{},\"input_config_dir\":\"/opt/ml/input/config\",\"input_data_config\":{\"images\":{\"RecordWrapperType\":\"None\",\"S3DistributionType\":\"FullyReplicated\",\"TrainingInputMode\":\"File\"}},\"input_dir\":\"/opt/ml/input\",\"instance_groups\":[\"homogeneousCluster\"],\"instance_groups_dict\":{\"homogeneousCluster\":{\"hosts\":[\"algo-1\"],\"instance_group_name\":\"homogeneousCluster\",\"instance_type\":\"ml.g5.2xlarge\"}},\"is_hetero\":false,\"is_master\":true,\"is_modelparallel_enabled\":null,\"is_smddpmprun_installed\":true,\"job_name\":\"sd-xl-dreambooth-finetuning-high-2023-08-25-10-08-30-083\",\"log_level\":20,\"master_hostname\":\"algo-1\",\"model_dir\":\"/opt/ml/model\",\"module_dir\":\"s3://sagemaker-us-west-2-687912291502/sd-xl-dreambooth-finetuning-high-2023-08-25-10-08-30-083/source/sourcedir.tar.gz\",\"module_name\":\"train.sh\",\"network_interface_name\":\"eth0\",\"num_cpus\":8,\"num_gpus\":1,\"num_neurons\":0,\"output_data_dir\":\"/opt/ml/output/data\",\"output_dir\":\"/opt/ml/output\",\"output_intermediate_dir\":\"/opt/ml/output/intermediate\",\"resource_config\":{\"current_group_name\":\"homogeneousCluster\",\"current_host\":\"algo-1\",\"current_instance_type\":\"ml.g5.2xlarge\",\"hosts\":[\"algo-1\"],\"instance_groups\":[{\"hosts\":[\"algo-1\"],\"instance_group_name\":\"homogeneousCluster\",\"instance_type\":\"ml.g5.2xlarge\"}],\"network_interface_name\":\"eth0\"},\"user_entry_point\":\"train.sh\"}\u001b[0m\n",
+      "\u001b[34mSM_USER_ARGS=[]\u001b[0m\n",
+      "\u001b[34mSM_OUTPUT_INTERMEDIATE_DIR=/opt/ml/output/intermediate\u001b[0m\n",
+      "\u001b[34mSM_CHANNEL_IMAGES=/opt/ml/input/data/images\u001b[0m\n",
+      "\u001b[34mPYTHONPATH=/opt/ml/code:/opt/conda/bin:/opt/conda/lib/python310.zip:/opt/conda/lib/python3.10:/opt/conda/lib/python3.10/lib-dynload:/opt/conda/lib/python3.10/site-packages\u001b[0m\n",
+      "\u001b[34mInvoking script with the following command:\u001b[0m\n",
+      "\u001b[34m/bin/sh -c \"./train.sh \"\u001b[0m\n",
+      "\u001b[34m2023-08-25 10:15:29,742 sagemaker-training-toolkit INFO     Exceptions not imported for SageMaker TF as Tensorflow is not installed.\u001b[0m\n",
+      "\u001b[34mtotal 12136\u001b[0m\n",
+      "\u001b[34m-rwxrwxrwx  1 1000 1000     1329 Aug 25 10:08 train.sh\u001b[0m\n",
+      "\u001b[34mdrwxrwxrwx 11 1000 1000     4096 Aug 25 02:24 diffusers\u001b[0m\n",
+      "\u001b[34m-rwxrwxrwx  1 1000 1000 12419072 Aug 25 02:10 s5cmd\u001b[0m\n",
+      "\u001b[34mObtaining file:///opt/ml/code/diffusers\u001b[0m\n",
+      "\u001b[34mInstalling build dependencies: started\u001b[0m\n",
+      "\u001b[34mInstalling build dependencies: finished with status 'done'\u001b[0m\n",
+      "\u001b[34mChecking if build backend supports build_editable: started\u001b[0m\n",
+      "\u001b[34mChecking if build backend supports build_editable: finished with status 'done'\u001b[0m\n",
+      "\u001b[34mGetting requirements to build editable: started\u001b[0m\n",
+      "\u001b[34mGetting requirements to build editable: finished with status 'done'\u001b[0m\n",
+      "\u001b[34mPreparing editable metadata (pyproject.toml): started\u001b[0m\n",
+      "\u001b[34mPreparing editable metadata (pyproject.toml): finished with status 'done'\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: importlib-metadata in /opt/conda/lib/python3.10/site-packages (from diffusers==0.21.0.dev0) (4.13.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: filelock in /opt/conda/lib/python3.10/site-packages (from diffusers==0.21.0.dev0) (3.12.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: huggingface-hub>=0.13.2 in /opt/conda/lib/python3.10/site-packages (from diffusers==0.21.0.dev0) (0.14.1)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: numpy in /opt/conda/lib/python3.10/site-packages (from diffusers==0.21.0.dev0) (1.23.5)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: regex!=2019.12.17 in /opt/conda/lib/python3.10/site-packages (from diffusers==0.21.0.dev0) (2023.5.5)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: requests in /opt/conda/lib/python3.10/site-packages (from diffusers==0.21.0.dev0) (2.28.2)\u001b[0m\n",
+      "\u001b[34mCollecting safetensors>=0.3.1 (from diffusers==0.21.0.dev0)\u001b[0m\n",
+      "\u001b[34mDownloading safetensors-0.3.3-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.3 MB)\u001b[0m\n",
+      "\u001b[34m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.3/1.3 MB 27.8 MB/s eta 0:00:00\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: Pillow in /opt/conda/lib/python3.10/site-packages (from diffusers==0.21.0.dev0) (9.4.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: fsspec in /opt/conda/lib/python3.10/site-packages (from huggingface-hub>=0.13.2->diffusers==0.21.0.dev0) (2023.5.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: tqdm>=4.42.1 in /opt/conda/lib/python3.10/site-packages (from huggingface-hub>=0.13.2->diffusers==0.21.0.dev0) (4.65.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: pyyaml>=5.1 in /opt/conda/lib/python3.10/site-packages (from huggingface-hub>=0.13.2->diffusers==0.21.0.dev0) (5.4.1)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: typing-extensions>=3.7.4.3 in /opt/conda/lib/python3.10/site-packages (from huggingface-hub>=0.13.2->diffusers==0.21.0.dev0) (4.5.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: packaging>=20.9 in /opt/conda/lib/python3.10/site-packages (from huggingface-hub>=0.13.2->diffusers==0.21.0.dev0) (23.1)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: zipp>=0.5 in /opt/conda/lib/python3.10/site-packages (from importlib-metadata->diffusers==0.21.0.dev0) (3.15.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: charset-normalizer<4,>=2 in /opt/conda/lib/python3.10/site-packages (from requests->diffusers==0.21.0.dev0) (3.1.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: idna<4,>=2.5 in /opt/conda/lib/python3.10/site-packages (from requests->diffusers==0.21.0.dev0) (3.4)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: urllib3<1.27,>=1.21.1 in /opt/conda/lib/python3.10/site-packages (from requests->diffusers==0.21.0.dev0) (1.26.15)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: certifi>=2017.4.17 in /opt/conda/lib/python3.10/site-packages (from requests->diffusers==0.21.0.dev0) (2023.5.7)\u001b[0m\n",
+      "\u001b[34mBuilding wheels for collected packages: diffusers\u001b[0m\n",
+      "\u001b[34mBuilding editable for diffusers (pyproject.toml): started\u001b[0m\n",
+      "\u001b[34mBuilding editable for diffusers (pyproject.toml): finished with status 'done'\u001b[0m\n",
+      "\u001b[34mCreated wheel for diffusers: filename=diffusers-0.21.0.dev0-0.editable-py3-none-any.whl size=10590 sha256=04174d74787363bfffd7777681648ee363934f287b5b9887e063e4bc9e6137b6\u001b[0m\n",
+      "\u001b[34mStored in directory: /tmp/pip-ephem-wheel-cache-94533sq4/wheels/e7/bd/4d/191eca0598c7d7deb715af4f393ac4f436cd03eb69d398c6e3\u001b[0m\n",
+      "\u001b[34mSuccessfully built diffusers\u001b[0m\n",
+      "\u001b[34mInstalling collected packages: safetensors, diffusers\u001b[0m\n",
+      "\u001b[34mAttempting uninstall: diffusers\u001b[0m\n",
+      "\u001b[34mFound existing installation: diffusers 0.16.1\u001b[0m\n",
+      "\u001b[34mUninstalling diffusers-0.16.1:\u001b[0m\n",
+      "\u001b[34mSuccessfully uninstalled diffusers-0.16.1\u001b[0m\n",
+      "\u001b[34mSuccessfully installed diffusers-0.21.0.dev0 safetensors-0.3.3\u001b[0m\n",
+      "\u001b[34mWARNING: 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\n",
+      "\u001b[34m[notice] A new release of pip is available: 23.1.2 -> 23.2.1\u001b[0m\n",
+      "\u001b[34m[notice] To update, run: pip install --upgrade pip\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: accelerate>=0.16.0 in /opt/conda/lib/python3.10/site-packages (from -r requirements_sdxl.txt (line 1)) (0.19.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: torchvision in /opt/conda/lib/python3.10/site-packages (from -r requirements_sdxl.txt (line 2)) (0.15.1)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: transformers>=4.25.1 in /opt/conda/lib/python3.10/site-packages (from -r requirements_sdxl.txt (line 3)) (4.28.1)\u001b[0m\n",
+      "\u001b[34mCollecting ftfy (from -r requirements_sdxl.txt (line 4))\u001b[0m\n",
+      "\u001b[34mDownloading ftfy-6.1.1-py3-none-any.whl (53 kB)\u001b[0m\n",
+      "\u001b[34m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 53.1/53.1 kB 3.8 MB/s eta 0:00:00\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: tensorboard in /opt/conda/lib/python3.10/site-packages (from -r requirements_sdxl.txt (line 5)) (2.13.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: Jinja2 in /opt/conda/lib/python3.10/site-packages (from -r requirements_sdxl.txt (line 6)) (3.1.2)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: numpy>=1.17 in /opt/conda/lib/python3.10/site-packages (from accelerate>=0.16.0->-r requirements_sdxl.txt (line 1)) (1.23.5)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: packaging>=20.0 in /opt/conda/lib/python3.10/site-packages (from accelerate>=0.16.0->-r requirements_sdxl.txt (line 1)) (23.1)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: psutil in /opt/conda/lib/python3.10/site-packages (from accelerate>=0.16.0->-r requirements_sdxl.txt (line 1)) (5.9.5)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: pyyaml in /opt/conda/lib/python3.10/site-packages (from accelerate>=0.16.0->-r requirements_sdxl.txt (line 1)) (5.4.1)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: torch>=1.6.0 in /opt/conda/lib/python3.10/site-packages (from accelerate>=0.16.0->-r requirements_sdxl.txt (line 1)) (2.0.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: requests in /opt/conda/lib/python3.10/site-packages (from torchvision->-r requirements_sdxl.txt (line 2)) (2.28.2)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: pillow!=8.3.*,>=5.3.0 in /opt/conda/lib/python3.10/site-packages (from torchvision->-r requirements_sdxl.txt (line 2)) (9.4.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: filelock in /opt/conda/lib/python3.10/site-packages (from transformers>=4.25.1->-r requirements_sdxl.txt (line 3)) (3.12.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: huggingface-hub<1.0,>=0.11.0 in /opt/conda/lib/python3.10/site-packages (from transformers>=4.25.1->-r requirements_sdxl.txt (line 3)) (0.14.1)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: regex!=2019.12.17 in /opt/conda/lib/python3.10/site-packages (from transformers>=4.25.1->-r requirements_sdxl.txt (line 3)) (2023.5.5)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: tokenizers!=0.11.3,<0.14,>=0.11.1 in /opt/conda/lib/python3.10/site-packages (from transformers>=4.25.1->-r requirements_sdxl.txt (line 3)) (0.13.3)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: tqdm>=4.27 in /opt/conda/lib/python3.10/site-packages (from transformers>=4.25.1->-r requirements_sdxl.txt (line 3)) (4.65.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: wcwidth>=0.2.5 in /opt/conda/lib/python3.10/site-packages (from ftfy->-r requirements_sdxl.txt (line 4)) (0.2.6)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: absl-py>=0.4 in /opt/conda/lib/python3.10/site-packages (from tensorboard->-r requirements_sdxl.txt (line 5)) (1.4.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: grpcio>=1.48.2 in /opt/conda/lib/python3.10/site-packages (from tensorboard->-r requirements_sdxl.txt (line 5)) (1.54.2)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: google-auth<3,>=1.6.3 in /opt/conda/lib/python3.10/site-packages (from tensorboard->-r requirements_sdxl.txt (line 5)) (2.18.1)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: google-auth-oauthlib<1.1,>=0.5 in /opt/conda/lib/python3.10/site-packages (from tensorboard->-r requirements_sdxl.txt (line 5)) (1.0.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: markdown>=2.6.8 in /opt/conda/lib/python3.10/site-packages (from tensorboard->-r requirements_sdxl.txt (line 5)) (3.4.3)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: protobuf>=3.19.6 in /opt/conda/lib/python3.10/site-packages (from tensorboard->-r requirements_sdxl.txt (line 5)) (3.20.2)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: setuptools>=41.0.0 in /opt/conda/lib/python3.10/site-packages (from tensorboard->-r requirements_sdxl.txt (line 5)) (65.6.3)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: tensorboard-data-server<0.8.0,>=0.7.0 in /opt/conda/lib/python3.10/site-packages (from tensorboard->-r requirements_sdxl.txt (line 5)) (0.7.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: werkzeug>=1.0.1 in /opt/conda/lib/python3.10/site-packages (from tensorboard->-r requirements_sdxl.txt (line 5)) (2.3.4)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: wheel>=0.26 in /opt/conda/lib/python3.10/site-packages (from tensorboard->-r requirements_sdxl.txt (line 5)) (0.40.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: MarkupSafe>=2.0 in /opt/conda/lib/python3.10/site-packages (from Jinja2->-r requirements_sdxl.txt (line 6)) (2.1.2)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: cachetools<6.0,>=2.0.0 in /opt/conda/lib/python3.10/site-packages (from google-auth<3,>=1.6.3->tensorboard->-r requirements_sdxl.txt (line 5)) (5.3.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: pyasn1-modules>=0.2.1 in /opt/conda/lib/python3.10/site-packages (from google-auth<3,>=1.6.3->tensorboard->-r requirements_sdxl.txt (line 5)) (0.3.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: six>=1.9.0 in /opt/conda/lib/python3.10/site-packages (from google-auth<3,>=1.6.3->tensorboard->-r requirements_sdxl.txt (line 5)) (1.16.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: urllib3<2.0 in /opt/conda/lib/python3.10/site-packages (from google-auth<3,>=1.6.3->tensorboard->-r requirements_sdxl.txt (line 5)) (1.26.15)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: rsa<5,>=3.1.4 in /opt/conda/lib/python3.10/site-packages (from google-auth<3,>=1.6.3->tensorboard->-r requirements_sdxl.txt (line 5)) (4.7.2)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: requests-oauthlib>=0.7.0 in /opt/conda/lib/python3.10/site-packages (from google-auth-oauthlib<1.1,>=0.5->tensorboard->-r requirements_sdxl.txt (line 5)) (1.3.1)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: fsspec in /opt/conda/lib/python3.10/site-packages (from huggingface-hub<1.0,>=0.11.0->transformers>=4.25.1->-r requirements_sdxl.txt (line 3)) (2023.5.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: typing-extensions>=3.7.4.3 in /opt/conda/lib/python3.10/site-packages (from huggingface-hub<1.0,>=0.11.0->transformers>=4.25.1->-r requirements_sdxl.txt (line 3)) (4.5.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: charset-normalizer<4,>=2 in /opt/conda/lib/python3.10/site-packages (from requests->torchvision->-r requirements_sdxl.txt (line 2)) (3.1.0)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: idna<4,>=2.5 in /opt/conda/lib/python3.10/site-packages (from requests->torchvision->-r requirements_sdxl.txt (line 2)) (3.4)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: certifi>=2017.4.17 in /opt/conda/lib/python3.10/site-packages (from requests->torchvision->-r requirements_sdxl.txt (line 2)) (2023.5.7)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: sympy in /opt/conda/lib/python3.10/site-packages (from torch>=1.6.0->accelerate>=0.16.0->-r requirements_sdxl.txt (line 1)) (1.11.1)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: networkx in /opt/conda/lib/python3.10/site-packages (from torch>=1.6.0->accelerate>=0.16.0->-r requirements_sdxl.txt (line 1)) (3.1)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: pyasn1<0.6.0,>=0.4.6 in /opt/conda/lib/python3.10/site-packages (from pyasn1-modules>=0.2.1->google-auth<3,>=1.6.3->tensorboard->-r requirements_sdxl.txt (line 5)) (0.4.8)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: oauthlib>=3.0.0 in /opt/conda/lib/python3.10/site-packages (from requests-oauthlib>=0.7.0->google-auth-oauthlib<1.1,>=0.5->tensorboard->-r requirements_sdxl.txt (line 5)) (3.2.2)\u001b[0m\n",
+      "\u001b[34mRequirement already satisfied: mpmath>=0.19 in /opt/conda/lib/python3.10/site-packages (from sympy->torch>=1.6.0->accelerate>=0.16.0->-r requirements_sdxl.txt (line 1)) (1.3.0)\u001b[0m\n",
+      "\u001b[34mInstalling collected packages: ftfy\u001b[0m\n",
+      "\u001b[34mSuccessfully installed ftfy-6.1.1\u001b[0m\n",
+      "\u001b[34mWARNING: 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\n",
+      "\u001b[34m[notice] A new release of pip is available: 23.1.2 -> 23.2.1\u001b[0m\n",
+      "\u001b[34m[notice] To update, run: pip install --upgrade pip\u001b[0m\n",
+      "\u001b[34m[10:15:42] WARNING  The following values were not passed to        launch.py:890\n",
+      "                    `accelerate launch` and had defaults used                   \n",
+      "                    instead:                                                    \n",
+      "                            `--num_processes` was set to a value                \n",
+      "                    of `1`                                                      \n",
+      "                            `--num_machines` was set to a value of              \n",
+      "                    `1`                                                         \n",
+      "                            `--mixed_precision` was set to a value              \n",
+      "                    of `'no'`                                                   \n",
+      "                            `--dynamo_backend` was set to a value               \n",
+      "                    of `'no'`                                                   \n",
+      "                    To avoid this warning pass in values for each               \n",
+      "                    of the problematic parameters or run                        \n",
+      "                    `accelerate config`.\u001b[0m\n",
+      "\u001b[34m08/25/2023 10:15:46 - INFO - __main__ - Distributed environment: NO\u001b[0m\n",
+      "\u001b[34mNum processes: 1\u001b[0m\n",
+      "\u001b[34mProcess index: 0\u001b[0m\n",
+      "\u001b[34mLocal process index: 0\u001b[0m\n",
+      "\u001b[34mDevice: cuda\u001b[0m\n",
+      "\u001b[34mMixed precision type: fp16\u001b[0m\n",
+      "\u001b[34mDownloading (…)okenizer_config.json:   0%|          | 0.00/737 [00:00<?, ?B/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)okenizer_config.json: 100%|██████████| 737/737 [00:00<00:00, 7.34MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)tokenizer/vocab.json:   0%|          | 0.00/1.06M [00:00<?, ?B/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)tokenizer/vocab.json: 100%|██████████| 1.06M/1.06M [00:00<00:00, 7.72MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)tokenizer/vocab.json: 100%|██████████| 1.06M/1.06M [00:00<00:00, 7.69MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)tokenizer/merges.txt:   0%|          | 0.00/525k [00:00<?, ?B/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)tokenizer/merges.txt: 100%|██████████| 525k/525k [00:00<00:00, 47.7MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)cial_tokens_map.json:   0%|          | 0.00/472 [00:00<?, ?B/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)cial_tokens_map.json: 100%|██████████| 472/472 [00:00<00:00, 6.41MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)okenizer_config.json:   0%|          | 0.00/725 [00:00<?, ?B/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)okenizer_config.json: 100%|██████████| 725/725 [00:00<00:00, 8.15MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)cial_tokens_map.json:   0%|          | 0.00/460 [00:00<?, ?B/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)cial_tokens_map.json: 100%|██████████| 460/460 [00:00<00:00, 5.81MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)_encoder/config.json:   0%|          | 0.00/565 [00:00<?, ?B/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)_encoder/config.json: 100%|██████████| 565/565 [00:00<00:00, 6.24MB/s]\u001b[0m\n",
+      "\u001b[34mYou are using a model of type clip_text_model to instantiate a model of type . This is not supported for all configurations of models and can yield errors.\u001b[0m\n",
+      "\u001b[34mDownloading (…)ncoder_2/config.json:   0%|          | 0.00/575 [00:00<?, ?B/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ncoder_2/config.json: 100%|██████████| 575/575 [00:00<00:00, 6.26MB/s]\u001b[0m\n",
+      "\u001b[34mYou are using a model of type clip_text_model to instantiate a model of type . This is not supported for all configurations of models and can yield errors.\u001b[0m\n",
+      "\u001b[34mDownloading (…)cheduler_config.json:   0%|          | 0.00/479 [00:00<?, ?B/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)cheduler_config.json: 100%|██████████| 479/479 [00:00<00:00, 6.36MB/s]\u001b[0m\n",
+      "\u001b[34m{'thresholding', 'clip_sample_range', 'dynamic_thresholding_ratio', 'variance_type'} was not found in config. Values will be initialized to default values.\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:   0%|          | 0.00/492M [00:00<?, ?B/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:   4%|▍         | 21.0M/492M [00:00<00:02, 191MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  11%|█         | 52.4M/492M [00:00<00:01, 240MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  19%|█▉        | 94.4M/492M [00:00<00:01, 303MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  28%|██▊       | 136M/492M [00:00<00:01, 337MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  36%|███▌      | 178M/492M [00:00<00:00, 353MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  45%|████▍     | 220M/492M [00:00<00:00, 370MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  53%|█████▎    | 262M/492M [00:00<00:00, 378MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  62%|██████▏   | 304M/492M [00:00<00:00, 384MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  70%|███████   | 346M/492M [00:00<00:00, 388MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  79%|███████▉  | 388M/492M [00:01<00:00, 390MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  87%|████████▋ | 430M/492M [00:01<00:00, 383MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  96%|█████████▌| 472M/492M [00:01<00:00, 388MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors: 100%|██████████| 492M/492M [00:01<00:00, 364MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:   0%|          | 0.00/2.78G [00:00<?, ?B/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:   1%|          | 21.0M/2.78G [00:00<00:14, 194MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:   2%|▏         | 52.4M/2.78G [00:00<00:11, 228MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:   3%|▎         | 94.4M/2.78G [00:00<00:09, 289MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:   5%|▍         | 136M/2.78G [00:00<00:08, 327MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:   6%|▋         | 178M/2.78G [00:00<00:07, 347MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:   8%|▊         | 220M/2.78G [00:00<00:07, 365MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:   9%|▉         | 262M/2.78G [00:00<00:06, 362MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  11%|█         | 304M/2.78G [00:00<00:06, 357MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  12%|█▏        | 346M/2.78G [00:01<00:06, 369MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  14%|█▍        | 388M/2.78G [00:01<00:06, 374MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  15%|█▌        | 430M/2.78G [00:01<00:06, 379MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  17%|█▋        | 472M/2.78G [00:01<00:05, 387MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  18%|█▊        | 514M/2.78G [00:01<00:05, 395MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  20%|██        | 556M/2.78G [00:01<00:05, 395MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  22%|██▏       | 598M/2.78G [00:01<00:05, 397MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  23%|██▎       | 640M/2.78G [00:01<00:05, 399MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  25%|██▍       | 682M/2.78G [00:01<00:05, 402MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  26%|██▌       | 724M/2.78G [00:01<00:05, 403MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  28%|██▊       | 765M/2.78G [00:02<00:04, 404MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  29%|██▉       | 807M/2.78G [00:02<00:04, 404MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  31%|███       | 849M/2.78G [00:02<00:04, 395MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  32%|███▏      | 891M/2.78G [00:02<00:04, 393MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  34%|███▎      | 933M/2.78G [00:02<00:04, 398MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  35%|███▌      | 975M/2.78G [00:02<00:04, 402MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  37%|███▋      | 1.02G/2.78G [00:02<00:04, 403MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  38%|███▊      | 1.06G/2.78G [00:02<00:04, 401MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  40%|███▉      | 1.10G/2.78G [00:02<00:04, 394MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  41%|████      | 1.14G/2.78G [00:03<00:04, 394MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  43%|████▎     | 1.18G/2.78G [00:03<00:04, 395MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  44%|████▍     | 1.23G/2.78G [00:03<00:03, 399MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  46%|████▌     | 1.27G/2.78G [00:03<00:03, 399MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  47%|████▋     | 1.31G/2.78G [00:03<00:03, 400MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  49%|████▊     | 1.35G/2.78G [00:03<00:03, 405MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  50%|█████     | 1.39G/2.78G [00:03<00:03, 401MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  52%|█████▏    | 1.44G/2.78G [00:03<00:03, 402MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  53%|█████▎    | 1.48G/2.78G [00:03<00:03, 403MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  55%|█████▍    | 1.52G/2.78G [00:03<00:03, 402MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  56%|█████▌    | 1.56G/2.78G [00:04<00:03, 403MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  58%|█████▊    | 1.60G/2.78G [00:04<00:02, 402MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  59%|█████▉    | 1.65G/2.78G [00:04<00:02, 399MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  61%|██████    | 1.69G/2.78G [00:04<00:02, 400MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  62%|██████▏   | 1.73G/2.78G [00:04<00:02, 400MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  64%|██████▍   | 1.77G/2.78G [00:04<00:02, 401MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  65%|██████▌   | 1.81G/2.78G [00:04<00:02, 400MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  67%|██████▋   | 1.86G/2.78G [00:04<00:02, 401MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  68%|██████▊   | 1.90G/2.78G [00:04<00:02, 398MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  70%|██████▉   | 1.94G/2.78G [00:05<00:02, 385MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  71%|███████▏  | 1.98G/2.78G [00:05<00:02, 387MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  73%|███████▎  | 2.02G/2.78G [00:05<00:01, 393MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  74%|███████▍  | 2.07G/2.78G [00:05<00:01, 392MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  76%|███████▌  | 2.11G/2.78G [00:05<00:01, 393MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  77%|███████▋  | 2.15G/2.78G [00:05<00:01, 398MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  79%|███████▉  | 2.19G/2.78G [00:05<00:01, 400MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  80%|████████  | 2.23G/2.78G [00:05<00:01, 399MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  82%|████████▏ | 2.28G/2.78G [00:05<00:01, 400MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  83%|████████▎ | 2.32G/2.78G [00:05<00:01, 402MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  85%|████████▍ | 2.36G/2.78G [00:06<00:01, 402MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  86%|████████▋ | 2.40G/2.78G [00:06<00:00, 404MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  88%|████████▊ | 2.44G/2.78G [00:06<00:00, 401MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  89%|████████▉ | 2.49G/2.78G [00:06<00:00, 403MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  91%|█████████ | 2.53G/2.78G [00:06<00:00, 404MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  92%|█████████▏| 2.57G/2.78G [00:06<00:00, 403MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  94%|█████████▍| 2.61G/2.78G [00:06<00:00, 405MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  95%|█████████▌| 2.65G/2.78G [00:06<00:00, 404MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  97%|█████████▋| 2.69G/2.78G [00:06<00:00, 404MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors:  98%|█████████▊| 2.74G/2.78G [00:06<00:00, 401MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors: 100%|██████████| 2.78G/2.78G [00:07<00:00, 403MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading model.safetensors: 100%|██████████| 2.78G/2.78G [00:07<00:00, 391MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)lve/main/config.json:   0%|          | 0.00/631 [00:00<?, ?B/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)lve/main/config.json: 100%|██████████| 631/631 [00:00<00:00, 7.33MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   0%|          | 0.00/335M [00:00<?, ?B/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   3%|▎         | 10.5M/335M [00:00<00:07, 43.6MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  16%|█▌        | 52.4M/335M [00:00<00:01, 168MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  28%|██▊       | 94.4M/335M [00:00<00:00, 247MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  41%|████      | 136M/335M [00:00<00:00, 294MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  53%|█████▎    | 178M/335M [00:00<00:00, 327MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  66%|██████▌   | 220M/335M [00:00<00:00, 348MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  78%|███████▊  | 262M/335M [00:00<00:00, 363MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  91%|█████████ | 304M/335M [00:01<00:00, 372MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors: 100%|██████████| 335M/335M [00:01<00:00, 310MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ain/unet/config.json:   0%|          | 0.00/1.68k [00:00<?, ?B/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ain/unet/config.json: 100%|██████████| 1.68k/1.68k [00:00<00:00, 16.3MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   0%|          | 0.00/10.3G [00:00<?, ?B/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   1%|          | 52.4M/10.3G [00:00<00:24, 413MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   1%|          | 94.4M/10.3G [00:00<00:26, 382MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   1%|▏         | 136M/10.3G [00:00<00:26, 378MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   2%|▏         | 178M/10.3G [00:00<00:26, 381MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   2%|▏         | 220M/10.3G [00:00<00:26, 386MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   3%|▎         | 262M/10.3G [00:00<00:25, 387MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   3%|▎         | 304M/10.3G [00:00<00:25, 389MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   3%|▎         | 346M/10.3G [00:00<00:25, 394MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   4%|▍         | 388M/10.3G [00:00<00:24, 400MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   4%|▍         | 430M/10.3G [00:01<00:24, 404MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   5%|▍         | 472M/10.3G [00:01<00:24, 405MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   5%|▌         | 514M/10.3G [00:01<00:24, 406MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   5%|▌         | 556M/10.3G [00:01<00:24, 403MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   6%|▌         | 598M/10.3G [00:01<00:24, 401MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   6%|▌         | 640M/10.3G [00:01<00:23, 403MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   7%|▋         | 682M/10.3G [00:01<00:23, 403MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   7%|▋         | 724M/10.3G [00:01<00:23, 407MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   7%|▋         | 765M/10.3G [00:01<00:23, 406MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   8%|▊         | 807M/10.3G [00:02<00:23, 406MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   8%|▊         | 849M/10.3G [00:02<00:23, 406MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   9%|▊         | 891M/10.3G [00:02<00:23, 402MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   9%|▉         | 933M/10.3G [00:02<00:23, 404MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:   9%|▉         | 975M/10.3G [00:02<00:22, 406MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  10%|▉         | 1.02G/10.3G [00:02<00:22, 403MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  10%|█         | 1.06G/10.3G [00:02<00:23, 400MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  11%|█         | 1.10G/10.3G [00:02<00:22, 399MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  11%|█         | 1.14G/10.3G [00:02<00:22, 399MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  12%|█▏        | 1.18G/10.3G [00:02<00:23, 392MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  12%|█▏        | 1.23G/10.3G [00:03<00:22, 394MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  12%|█▏        | 1.27G/10.3G [00:03<00:22, 396MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  13%|█▎        | 1.31G/10.3G [00:03<00:22, 394MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  13%|█▎        | 1.35G/10.3G [00:03<00:22, 393MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  14%|█▎        | 1.39G/10.3G [00:03<00:22, 394MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  14%|█▍        | 1.44G/10.3G [00:03<00:22, 398MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  14%|█▍        | 1.48G/10.3G [00:03<00:22, 394MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  15%|█▍        | 1.52G/10.3G [00:03<00:22, 385MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  15%|█▌        | 1.56G/10.3G [00:03<00:22, 390MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  16%|█▌        | 1.60G/10.3G [00:04<00:22, 394MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  16%|█▌        | 1.65G/10.3G [00:04<00:21, 394MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  16%|█▋        | 1.69G/10.3G [00:04<00:22, 379MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  17%|█▋        | 1.73G/10.3G [00:04<00:22, 382MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  17%|█▋        | 1.77G/10.3G [00:04<00:21, 388MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  18%|█▊        | 1.81G/10.3G [00:04<00:21, 389MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  18%|█▊        | 1.86G/10.3G [00:04<00:21, 391MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  18%|█▊        | 1.90G/10.3G [00:04<00:25, 332MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  19%|█▉        | 1.94G/10.3G [00:04<00:23, 347MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  19%|█▉        | 1.98G/10.3G [00:05<00:22, 362MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  20%|█▉        | 2.02G/10.3G [00:05<00:22, 373MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  20%|██        | 2.07G/10.3G [00:05<00:21, 381MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  21%|██        | 2.11G/10.3G [00:05<00:21, 383MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  21%|██        | 2.15G/10.3G [00:05<00:21, 385MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  21%|██▏       | 2.19G/10.3G [00:05<00:20, 390MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  22%|██▏       | 2.23G/10.3G [00:05<00:20, 394MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  22%|██▏       | 2.28G/10.3G [00:05<00:20, 394MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  23%|██▎       | 2.32G/10.3G [00:05<00:20, 385MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  23%|██▎       | 2.36G/10.3G [00:06<00:20, 389MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  23%|██▎       | 2.40G/10.3G [00:06<00:20, 384MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  24%|██▍       | 2.44G/10.3G [00:06<00:20, 384MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  24%|██▍       | 2.49G/10.3G [00:06<00:20, 387MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  25%|██▍       | 2.53G/10.3G [00:06<00:20, 369MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  25%|██▌       | 2.57G/10.3G [00:06<00:20, 372MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  25%|██▌       | 2.61G/10.3G [00:06<00:20, 380MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  26%|██▌       | 2.65G/10.3G [00:06<00:19, 386MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  26%|██▌       | 2.69G/10.3G [00:06<00:21, 350MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  27%|██▋       | 2.74G/10.3G [00:07<00:21, 356MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  27%|██▋       | 2.78G/10.3G [00:07<00:20, 365MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  27%|██▋       | 2.82G/10.3G [00:07<00:20, 369MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  28%|██▊       | 2.86G/10.3G [00:07<00:19, 377MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  28%|██▊       | 2.90G/10.3G [00:07<00:19, 381MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  29%|██▊       | 2.95G/10.3G [00:07<00:18, 388MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  29%|██▉       | 2.99G/10.3G [00:07<00:18, 393MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  30%|██▉       | 3.03G/10.3G [00:07<00:18, 389MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  30%|██▉       | 3.07G/10.3G [00:07<00:18, 390MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  30%|███       | 3.11G/10.3G [00:08<00:19, 372MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  31%|███       | 3.16G/10.3G [00:08<00:21, 328MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  31%|███       | 3.20G/10.3G [00:08<00:24, 293MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  31%|███▏      | 3.23G/10.3G [00:08<00:25, 276MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  32%|███▏      | 3.26G/10.3G [00:08<00:26, 265MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  32%|███▏      | 3.29G/10.3G [00:08<00:27, 258MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  32%|███▏      | 3.32G/10.3G [00:08<00:27, 252MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  33%|███▎      | 3.36G/10.3G [00:09<00:27, 248MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  33%|███▎      | 3.39G/10.3G [00:09<00:28, 244MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  33%|███▎      | 3.42G/10.3G [00:09<00:28, 243MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  34%|███▎      | 3.45G/10.3G [00:09<00:27, 244MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  34%|███▍      | 3.48G/10.3G [00:09<00:28, 240MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  34%|███▍      | 3.51G/10.3G [00:09<00:28, 239MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  35%|███▍      | 3.54G/10.3G [00:09<00:28, 235MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  35%|███▍      | 3.58G/10.3G [00:09<00:28, 238MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  35%|███▌      | 3.61G/10.3G [00:10<00:27, 241MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  35%|███▌      | 3.64G/10.3G [00:10<00:28, 237MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  36%|███▌      | 3.67G/10.3G [00:10<00:27, 240MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  36%|███▌      | 3.70G/10.3G [00:10<00:27, 237MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  36%|███▋      | 3.73G/10.3G [00:10<00:27, 238MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  37%|███▋      | 3.76G/10.3G [00:10<00:27, 238MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  37%|███▋      | 3.80G/10.3G [00:10<00:26, 240MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  37%|███▋      | 3.83G/10.3G [00:11<00:26, 239MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  38%|███▊      | 3.86G/10.3G [00:11<00:27, 237MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  38%|███▊      | 3.89G/10.3G [00:11<00:26, 238MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  38%|███▊      | 3.92G/10.3G [00:11<00:26, 238MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  38%|███▊      | 3.95G/10.3G [00:11<00:33, 190MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  39%|███▉      | 3.98G/10.3G [00:11<00:32, 194MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  39%|███▉      | 4.01G/10.3G [00:12<00:36, 171MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  39%|███▉      | 4.03G/10.3G [00:12<00:40, 155MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  39%|███▉      | 4.05G/10.3G [00:12<00:38, 160MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  40%|███▉      | 4.07G/10.3G [00:12<00:42, 147MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  40%|███▉      | 4.09G/10.3G [00:12<00:44, 138MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  40%|████      | 4.11G/10.3G [00:12<00:45, 137MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  40%|████      | 4.13G/10.3G [00:12<00:46, 132MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  40%|████      | 4.15G/10.3G [00:13<00:48, 126MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  41%|████      | 4.17G/10.3G [00:13<00:47, 127MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  41%|████      | 4.19G/10.3G [00:13<00:49, 124MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  41%|████      | 4.22G/10.3G [00:13<00:49, 123MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  41%|████      | 4.24G/10.3G [00:13<00:50, 120MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  41%|████▏     | 4.26G/10.3G [00:14<00:49, 121MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  42%|████▏     | 4.28G/10.3G [00:14<00:49, 120MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  42%|████▏     | 4.30G/10.3G [00:14<00:58, 102MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  42%|████▏     | 4.33G/10.3G [00:14<00:47, 124MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  42%|████▏     | 4.35G/10.3G [00:14<00:47, 124MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  43%|████▎     | 4.37G/10.3G [00:14<00:47, 125MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  43%|████▎     | 4.39G/10.3G [00:15<00:47, 123MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  43%|████▎     | 4.41G/10.3G [00:15<00:48, 121MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  43%|████▎     | 4.44G/10.3G [00:15<00:47, 122MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  43%|████▎     | 4.46G/10.3G [00:15<00:47, 121MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  44%|████▎     | 4.48G/10.3G [00:15<00:49, 118MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  44%|████▍     | 4.50G/10.3G [00:16<00:48, 120MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  44%|████▍     | 4.52G/10.3G [00:16<00:47, 120MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  44%|████▍     | 4.54G/10.3G [00:16<00:47, 120MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  44%|████▍     | 4.56G/10.3G [00:16<00:47, 120MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  45%|████▍     | 4.58G/10.3G [00:16<00:47, 119MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  45%|████▍     | 4.60G/10.3G [00:16<00:46, 123MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  45%|████▌     | 4.62G/10.3G [00:17<00:47, 120MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  45%|████▌     | 4.65G/10.3G [00:17<00:46, 121MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  45%|████▌     | 4.67G/10.3G [00:17<00:46, 120MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  46%|████▌     | 4.69G/10.3G [00:17<00:46, 120MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  46%|████▌     | 4.71G/10.3G [00:17<00:45, 121MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  46%|████▌     | 4.73G/10.3G [00:17<00:45, 121MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  46%|████▋     | 4.75G/10.3G [00:18<00:45, 120MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  46%|████▋     | 4.77G/10.3G [00:18<00:43, 125MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  47%|████▋     | 4.79G/10.3G [00:18<00:43, 125MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  47%|████▋     | 4.81G/10.3G [00:18<00:44, 123MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  47%|████▋     | 4.83G/10.3G [00:18<00:44, 123MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  47%|████▋     | 4.85G/10.3G [00:18<00:44, 122MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  47%|████▋     | 4.88G/10.3G [00:19<00:44, 121MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  48%|████▊     | 4.90G/10.3G [00:19<00:41, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  48%|████▊     | 4.92G/10.3G [00:19<00:42, 126MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  48%|████▊     | 4.94G/10.3G [00:19<00:43, 123MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  48%|████▊     | 4.96G/10.3G [00:19<00:41, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  48%|████▊     | 4.98G/10.3G [00:19<00:42, 126MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  49%|████▊     | 5.00G/10.3G [00:20<00:42, 124MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  49%|████▉     | 5.02G/10.3G [00:20<00:40, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  49%|████▉     | 5.04G/10.3G [00:20<00:42, 124MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  49%|████▉     | 5.06G/10.3G [00:20<00:42, 123MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  50%|████▉     | 5.09G/10.3G [00:20<00:39, 131MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  50%|████▉     | 5.11G/10.3G [00:20<00:40, 126MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  50%|████▉     | 5.13G/10.3G [00:21<00:41, 124MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  50%|█████     | 5.15G/10.3G [00:21<00:39, 131MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  50%|█████     | 5.17G/10.3G [00:21<00:39, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  51%|█████     | 5.19G/10.3G [00:21<00:40, 127MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  51%|█████     | 5.21G/10.3G [00:21<00:38, 132MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  51%|█████     | 5.23G/10.3G [00:21<00:37, 133MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  51%|█████     | 5.25G/10.3G [00:22<00:40, 125MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  51%|█████▏    | 5.27G/10.3G [00:22<00:38, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  52%|█████▏    | 5.30G/10.3G [00:22<00:38, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  52%|█████▏    | 5.32G/10.3G [00:22<00:39, 126MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  52%|█████▏    | 5.34G/10.3G [00:22<00:38, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  52%|█████▏    | 5.36G/10.3G [00:22<00:39, 125MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  52%|█████▏    | 5.38G/10.3G [00:23<00:39, 124MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  53%|█████▎    | 5.40G/10.3G [00:23<00:38, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  53%|█████▎    | 5.42G/10.3G [00:23<00:36, 131MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  53%|█████▎    | 5.44G/10.3G [00:23<00:37, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  53%|█████▎    | 5.46G/10.3G [00:23<00:38, 126MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  53%|█████▎    | 5.48G/10.3G [00:23<00:36, 131MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  54%|█████▎    | 5.51G/10.3G [00:24<00:37, 127MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  54%|█████▍    | 5.53G/10.3G [00:24<00:36, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  54%|█████▍    | 5.55G/10.3G [00:24<00:36, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  54%|█████▍    | 5.57G/10.3G [00:24<00:36, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  54%|█████▍    | 5.59G/10.3G [00:24<00:37, 126MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  55%|█████▍    | 5.61G/10.3G [00:24<00:34, 134MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  55%|█████▍    | 5.63G/10.3G [00:25<00:35, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  55%|█████▌    | 5.65G/10.3G [00:25<00:36, 126MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  55%|█████▌    | 5.67G/10.3G [00:25<00:32, 142MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  55%|█████▌    | 5.69G/10.3G [00:25<00:34, 133MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  56%|█████▌    | 5.71G/10.3G [00:25<00:35, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  56%|█████▌    | 5.74G/10.3G [00:25<00:32, 138MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  56%|█████▌    | 5.76G/10.3G [00:25<00:34, 132MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  56%|█████▋    | 5.78G/10.3G [00:26<00:35, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  56%|█████▋    | 5.80G/10.3G [00:26<00:33, 132MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  57%|█████▋    | 5.82G/10.3G [00:26<00:34, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  57%|█████▋    | 5.84G/10.3G [00:26<00:35, 125MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  57%|█████▋    | 5.86G/10.3G [00:26<00:34, 127MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  57%|█████▋    | 5.88G/10.3G [00:26<00:33, 132MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  57%|█████▋    | 5.90G/10.3G [00:27<00:34, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  58%|█████▊    | 5.92G/10.3G [00:27<00:32, 133MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  58%|█████▊    | 5.95G/10.3G [00:27<00:33, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  58%|█████▊    | 5.97G/10.3G [00:27<00:33, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  58%|█████▊    | 5.99G/10.3G [00:27<00:32, 134MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  59%|█████▊    | 6.01G/10.3G [00:27<00:30, 138MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  59%|█████▊    | 6.03G/10.3G [00:28<00:32, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  59%|█████▉    | 6.05G/10.3G [00:28<00:32, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  59%|█████▉    | 6.07G/10.3G [00:28<00:31, 135MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  59%|█████▉    | 6.09G/10.3G [00:28<00:31, 132MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  60%|█████▉    | 6.11G/10.3G [00:28<00:32, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  60%|█████▉    | 6.13G/10.3G [00:28<00:29, 140MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  60%|█████▉    | 6.16G/10.3G [00:29<00:31, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  60%|██████    | 6.18G/10.3G [00:29<00:32, 126MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  60%|██████    | 6.20G/10.3G [00:29<00:28, 142MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  61%|██████    | 6.22G/10.3G [00:29<00:30, 134MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  61%|██████    | 6.24G/10.3G [00:29<00:31, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  61%|██████    | 6.26G/10.3G [00:29<00:29, 137MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  61%|██████    | 6.28G/10.3G [00:30<00:40, 99.6MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  61%|██████▏   | 6.31G/10.3G [00:30<00:29, 133MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  62%|██████▏   | 6.34G/10.3G [00:30<00:24, 157MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  62%|██████▏   | 6.36G/10.3G [00:30<00:26, 145MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  62%|██████▏   | 6.39G/10.3G [00:30<00:28, 138MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  62%|██████▏   | 6.41G/10.3G [00:30<00:27, 140MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  63%|██████▎   | 6.43G/10.3G [00:31<00:28, 134MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  63%|██████▎   | 6.45G/10.3G [00:31<00:29, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  63%|██████▎   | 6.47G/10.3G [00:31<00:28, 134MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  63%|██████▎   | 6.49G/10.3G [00:31<00:29, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  63%|██████▎   | 6.51G/10.3G [00:31<00:30, 124MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  64%|██████▎   | 6.53G/10.3G [00:31<00:27, 137MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  64%|██████▍   | 6.55G/10.3G [00:32<00:28, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  64%|██████▍   | 6.57G/10.3G [00:32<00:28, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  64%|██████▍   | 6.60G/10.3G [00:32<00:25, 142MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  64%|██████▍   | 6.62G/10.3G [00:32<00:27, 134MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  65%|██████▍   | 6.64G/10.3G [00:32<00:28, 127MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  65%|██████▍   | 6.66G/10.3G [00:32<00:25, 140MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  65%|██████▌   | 6.68G/10.3G [00:32<00:27, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  65%|██████▌   | 6.70G/10.3G [00:33<00:27, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  65%|██████▌   | 6.72G/10.3G [00:33<00:26, 135MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  66%|██████▌   | 6.74G/10.3G [00:33<00:26, 133MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  66%|██████▌   | 6.76G/10.3G [00:33<00:28, 124MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  66%|██████▌   | 6.78G/10.3G [00:33<00:26, 133MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  66%|██████▋   | 6.81G/10.3G [00:33<00:25, 134MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  66%|██████▋   | 6.83G/10.3G [00:34<00:26, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  67%|██████▋   | 6.85G/10.3G [00:34<00:25, 134MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  67%|██████▋   | 6.87G/10.3G [00:34<00:24, 138MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  67%|██████▋   | 6.89G/10.3G [00:34<00:25, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  67%|██████▋   | 6.91G/10.3G [00:34<00:25, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  67%|██████▋   | 6.93G/10.3G [00:34<00:23, 141MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  68%|██████▊   | 6.95G/10.3G [00:35<00:24, 134MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  68%|██████▊   | 6.97G/10.3G [00:35<00:25, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  68%|██████▊   | 6.99G/10.3G [00:35<00:22, 143MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  68%|██████▊   | 7.01G/10.3G [00:35<00:24, 134MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  69%|██████▊   | 7.04G/10.3G [00:35<00:25, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  69%|██████▊   | 7.06G/10.3G [00:35<00:23, 138MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  69%|██████▉   | 7.08G/10.3G [00:35<00:24, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  69%|██████▉   | 7.10G/10.3G [00:36<00:24, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  69%|██████▉   | 7.12G/10.3G [00:36<00:22, 138MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  70%|██████▉   | 7.14G/10.3G [00:36<00:23, 134MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  70%|██████▉   | 7.16G/10.3G [00:36<00:23, 132MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  70%|██████▉   | 7.18G/10.3G [00:36<00:22, 136MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  70%|███████   | 7.20G/10.3G [00:36<00:22, 139MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  70%|███████   | 7.22G/10.3G [00:37<00:22, 133MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  71%|███████   | 7.25G/10.3G [00:37<00:23, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  71%|███████   | 7.27G/10.3G [00:37<00:21, 138MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  71%|███████   | 7.29G/10.3G [00:37<00:22, 131MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  71%|███████   | 7.31G/10.3G [00:37<00:23, 126MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  71%|███████▏  | 7.33G/10.3G [00:37<00:20, 142MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  72%|███████▏  | 7.35G/10.3G [00:38<00:21, 134MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  72%|███████▏  | 7.37G/10.3G [00:38<00:22, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  72%|███████▏  | 7.39G/10.3G [00:38<00:20, 138MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  72%|███████▏  | 7.41G/10.3G [00:38<00:21, 131MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  72%|███████▏  | 7.43G/10.3G [00:38<00:21, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  73%|███████▎  | 7.46G/10.3G [00:38<00:20, 138MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  73%|███████▎  | 7.48G/10.3G [00:38<00:20, 136MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  73%|███████▎  | 7.50G/10.3G [00:39<00:21, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  73%|███████▎  | 7.52G/10.3G [00:39<00:20, 136MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  73%|███████▎  | 7.54G/10.3G [00:39<00:19, 138MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  74%|███████▎  | 7.56G/10.3G [00:39<00:20, 132MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  74%|███████▍  | 7.58G/10.3G [00:39<00:20, 131MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  74%|███████▍  | 7.60G/10.3G [00:39<00:19, 138MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  74%|███████▍  | 7.62G/10.3G [00:40<00:20, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  74%|███████▍  | 7.64G/10.3G [00:40<00:25, 101MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  75%|███████▍  | 7.69G/10.3G [00:40<00:17, 151MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  75%|███████▌  | 7.71G/10.3G [00:40<00:17, 144MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  75%|███████▌  | 7.73G/10.3G [00:40<00:17, 148MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  75%|███████▌  | 7.75G/10.3G [00:40<00:18, 140MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  76%|███████▌  | 7.77G/10.3G [00:41<00:18, 133MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  76%|███████▌  | 7.79G/10.3G [00:41<00:17, 141MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  76%|███████▌  | 7.81G/10.3G [00:41<00:18, 133MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  76%|███████▋  | 7.83G/10.3G [00:41<00:18, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  76%|███████▋  | 7.85G/10.3G [00:41<00:17, 137MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  77%|███████▋  | 7.87G/10.3G [00:41<00:17, 137MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  77%|███████▋  | 7.90G/10.3G [00:42<00:18, 131MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  77%|███████▋  | 7.92G/10.3G [00:42<00:17, 131MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  77%|███████▋  | 7.94G/10.3G [00:42<00:16, 137MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  77%|███████▋  | 7.96G/10.3G [00:42<00:17, 131MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  78%|███████▊  | 7.98G/10.3G [00:42<00:17, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  78%|███████▊  | 8.00G/10.3G [00:42<00:16, 136MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  78%|███████▊  | 8.02G/10.3G [00:43<00:17, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  78%|███████▊  | 8.04G/10.3G [00:43<00:17, 127MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  79%|███████▊  | 8.06G/10.3G [00:43<00:15, 142MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  79%|███████▊  | 8.08G/10.3G [00:43<00:15, 137MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  79%|███████▉  | 8.11G/10.3G [00:43<00:16, 131MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  79%|███████▉  | 8.13G/10.3G [00:43<00:15, 140MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  79%|███████▉  | 8.15G/10.3G [00:43<00:15, 135MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  80%|███████▉  | 8.17G/10.3G [00:44<00:15, 132MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  80%|███████▉  | 8.19G/10.3G [00:44<00:15, 137MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  80%|███████▉  | 8.21G/10.3G [00:44<00:15, 137MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  80%|████████  | 8.23G/10.3G [00:44<00:15, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  80%|████████  | 8.25G/10.3G [00:44<00:15, 132MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  81%|████████  | 8.27G/10.3G [00:44<00:14, 138MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  81%|████████  | 8.29G/10.3G [00:45<00:15, 132MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  81%|████████  | 8.32G/10.3G [00:45<00:15, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  81%|████████  | 8.34G/10.3G [00:45<00:14, 138MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  81%|████████▏ | 8.36G/10.3G [00:45<00:14, 132MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  82%|████████▏ | 8.38G/10.3G [00:45<00:14, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  82%|████████▏ | 8.40G/10.3G [00:45<00:13, 143MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  82%|████████▏ | 8.42G/10.3G [00:45<00:13, 137MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  82%|████████▏ | 8.44G/10.3G [00:46<00:13, 132MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  82%|████████▏ | 8.46G/10.3G [00:46<00:12, 141MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  83%|████████▎ | 8.48G/10.3G [00:46<00:13, 133MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  83%|████████▎ | 8.50G/10.3G [00:46<00:13, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  83%|████████▎ | 8.52G/10.3G [00:46<00:12, 135MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  83%|████████▎ | 8.55G/10.3G [00:46<00:12, 137MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  83%|████████▎ | 8.57G/10.3G [00:47<00:13, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  84%|████████▎ | 8.59G/10.3G [00:47<00:12, 133MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  84%|████████▍ | 8.61G/10.3G [00:47<00:12, 137MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  84%|████████▍ | 8.63G/10.3G [00:47<00:12, 131MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  84%|████████▍ | 8.65G/10.3G [00:47<00:12, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  84%|████████▍ | 8.67G/10.3G [00:47<00:11, 138MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  85%|████████▍ | 8.69G/10.3G [00:48<00:12, 131MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  85%|████████▍ | 8.71G/10.3G [00:48<00:12, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  85%|████████▌ | 8.73G/10.3G [00:48<00:10, 145MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  85%|████████▌ | 8.76G/10.3G [00:48<00:11, 135MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  85%|████████▌ | 8.78G/10.3G [00:48<00:11, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  86%|████████▌ | 8.80G/10.3G [00:48<00:10, 141MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  86%|████████▌ | 8.82G/10.3G [00:48<00:10, 134MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  86%|████████▌ | 8.84G/10.3G [00:49<00:11, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  86%|████████▋ | 8.86G/10.3G [00:49<00:10, 138MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  86%|████████▋ | 8.88G/10.3G [00:49<00:10, 136MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  87%|████████▋ | 8.90G/10.3G [00:49<00:10, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  87%|████████▋ | 8.92G/10.3G [00:49<00:09, 137MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  87%|████████▋ | 8.94G/10.3G [00:49<00:09, 143MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  87%|████████▋ | 8.97G/10.3G [00:50<00:09, 133MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  87%|████████▋ | 8.99G/10.3G [00:50<00:09, 132MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  88%|████████▊ | 9.01G/10.3G [00:50<00:09, 139MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  88%|████████▊ | 9.03G/10.3G [00:50<00:09, 133MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  88%|████████▊ | 9.05G/10.3G [00:50<00:09, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  88%|████████▊ | 9.07G/10.3G [00:50<00:08, 145MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  89%|████████▊ | 9.09G/10.3G [00:50<00:08, 137MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  89%|████████▊ | 9.11G/10.3G [00:51<00:08, 131MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  89%|████████▉ | 9.13G/10.3G [00:51<00:08, 140MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  89%|████████▉ | 9.15G/10.3G [00:51<00:08, 136MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  89%|████████▉ | 9.18G/10.3G [00:51<00:08, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  90%|████████▉ | 9.20G/10.3G [00:51<00:08, 134MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  90%|████████▉ | 9.22G/10.3G [00:51<00:07, 135MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  90%|████████▉ | 9.24G/10.3G [00:52<00:07, 132MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  90%|█████████ | 9.26G/10.3G [00:52<00:07, 132MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  90%|█████████ | 9.28G/10.3G [00:52<00:07, 135MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  91%|█████████ | 9.30G/10.3G [00:52<00:07, 128MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  91%|█████████ | 9.32G/10.3G [00:52<00:07, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  91%|█████████ | 9.34G/10.3G [00:52<00:06, 141MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  91%|█████████ | 9.36G/10.3G [00:53<00:06, 133MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  91%|█████████▏| 9.38G/10.3G [00:53<00:06, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  92%|█████████▏| 9.41G/10.3G [00:53<00:06, 142MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  92%|█████████▏| 9.43G/10.3G [00:53<00:06, 135MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  92%|█████████▏| 9.45G/10.3G [00:53<00:06, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  92%|█████████▏| 9.47G/10.3G [00:53<00:05, 143MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  92%|█████████▏| 9.49G/10.3G [00:53<00:05, 135MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  93%|█████████▎| 9.51G/10.3G [00:54<00:05, 130MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  93%|█████████▎| 9.53G/10.3G [00:54<00:05, 134MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  93%|█████████▎| 9.55G/10.3G [00:54<00:06, 108MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  93%|█████████▎| 9.58G/10.3G [00:54<00:05, 137MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  94%|█████████▎| 9.60G/10.3G [00:54<00:04, 145MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  94%|█████████▎| 9.63G/10.3G [00:54<00:04, 139MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  94%|█████████▍| 9.65G/10.3G [00:55<00:04, 133MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  94%|█████████▍| 9.67G/10.3G [00:55<00:04, 142MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  94%|█████████▍| 9.69G/10.3G [00:55<00:04, 138MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  95%|█████████▍| 9.71G/10.3G [00:55<00:04, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  95%|█████████▍| 9.73G/10.3G [00:55<00:04, 129MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  95%|█████████▍| 9.75G/10.3G [00:55<00:04, 126MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  95%|█████████▌| 9.77G/10.3G [00:56<00:04, 122MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  95%|█████████▌| 9.79G/10.3G [00:56<00:03, 121MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  96%|█████████▌| 9.81G/10.3G [00:56<00:03, 119MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  96%|█████████▌| 9.84G/10.3G [00:56<00:03, 119MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  96%|█████████▌| 9.86G/10.3G [00:56<00:03, 119MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  96%|█████████▌| 9.88G/10.3G [00:57<00:03, 119MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  96%|█████████▋| 9.90G/10.3G [00:57<00:03, 119MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  97%|█████████▋| 9.92G/10.3G [00:57<00:02, 119MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  97%|█████████▋| 9.94G/10.3G [00:57<00:02, 119MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  97%|█████████▋| 9.96G/10.3G [00:57<00:02, 121MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  97%|█████████▋| 9.98G/10.3G [00:57<00:02, 120MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  97%|█████████▋| 10.0G/10.3G [00:58<00:02, 120MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  98%|█████████▊| 10.0G/10.3G [00:58<00:02, 120MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  98%|█████████▊| 10.0G/10.3G [00:58<00:01, 119MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  98%|█████████▊| 10.1G/10.3G [00:58<00:01, 119MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  98%|█████████▊| 10.1G/10.3G [00:58<00:01, 119MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  98%|█████████▊| 10.1G/10.3G [00:58<00:01, 117MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  99%|█████████▊| 10.1G/10.3G [00:59<00:01, 118MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  99%|█████████▉| 10.2G/10.3G [00:59<00:00, 131MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  99%|█████████▉| 10.2G/10.3G [00:59<00:00, 131MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  99%|█████████▉| 10.2G/10.3G [00:59<00:00, 127MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors:  99%|█████████▉| 10.2G/10.3G [00:59<00:00, 136MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors: 100%|█████████▉| 10.2G/10.3G [00:59<00:00, 146MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors: 100%|█████████▉| 10.3G/10.3G [01:00<00:00, 137MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors: 100%|██████████| 10.3G/10.3G [01:00<00:00, 123MB/s]\u001b[0m\n",
+      "\u001b[34mDownloading (…)ch_model.safetensors: 100%|██████████| 10.3G/10.3G [01:00<00:00, 171MB/s]\u001b[0m\n",
+      "\u001b[34m{'attention_type'} was not found in config. Values will be initialized to default values.\u001b[0m\n",
+      "\u001b[34m08/25/2023 10:17:17 - INFO - __main__ - ***** Running training *****\u001b[0m\n",
+      "\u001b[34m08/25/2023 10:17:17 - INFO - __main__ -   Num examples = 5\u001b[0m\n",
+      "\u001b[34m08/25/2023 10:17:17 - INFO - __main__ -   Num batches each epoch = 5\u001b[0m\n",
+      "\u001b[34m08/25/2023 10:17:17 - INFO - __main__ -   Num Epochs = 250\u001b[0m\n",
+      "\u001b[34m08/25/2023 10:17:17 - INFO - __main__ -   Instantaneous batch size per device = 1\u001b[0m\n",
+      "\u001b[34m08/25/2023 10:17:17 - INFO - __main__ -   Total train batch size (w. parallel, distributed & accumulation) = 4\u001b[0m\n",
+      "\u001b[34m08/25/2023 10:17:17 - INFO - __main__ -   Gradient Accumulation steps = 4\u001b[0m\n",
+      "\u001b[34m08/25/2023 10:17:17 - INFO - __main__ -   Total optimization steps = 500\u001b[0m\n",
+      "\u001b[34m0%|          | 0/500 [00:00<?, ?it/s]\u001b[0m\n",
+      "\u001b[34mSteps:   0%|          | 0/500 [00:00<?, ?it/s]\u001b[0m\n",
+      "\u001b[34mSteps:   0%|          | 0/500 [00:02<?, ?it/s, loss=0.00433, lr=1e-5]\u001b[0m\n",
+      "\u001b[34mSteps:   0%|          | 0/500 [00:04<?, ?it/s, loss=0.0569, lr=1e-5]\u001b[0m\n",
+      "\u001b[34mSteps:   0%|          | 0/500 [00:05<?, ?it/s, loss=0.293, lr=1e-5]\u001b[0m\n",
+      "\u001b[34mSteps:   0%|          | 1/500 [00:07<1:04:57,  7.81s/it, loss=0.293, lr=1e-5]\u001b[0m\n",
+      "\u001b[34mSteps:   0%|          | 1/500 [00:07<1:04:57,  7.81s/it, loss=0.0123, lr=1e-5]\u001b[0m\n",
+      "\u001b[34mSteps:   0%|          | 2/500 [00:09<34:27,  4.15s/it, loss=0.0123, lr=1e-5]\u001b[0m\n",
+      "\u001b[34mSteps:   0%|          | 2/500 [00:09<34:27,  4.15s/it, loss=0.0188, lr=1e-5]\u001b[0m\n",
+      "\u001b[34m08/25/2023 10:17:27 - INFO - __main__ - Running validation... \n",
+      " Generating 4 images with prompt: A photo of sks dog in a bucket.\u001b[0m\n"
+     ]
+    }
+   ],
+   "source": [
+    "import time\n",
+    "from sagemaker.estimator import Estimator\n",
+    "from sagemaker.pytorch.estimator import PyTorch\n",
+    "\n",
+    "environment = {\n",
+    "    'PYTORCH_CUDA_ALLOC_CONF':'max_split_size_mb:32'\n",
+    "}\n",
+    "\n",
+    "## The image uri which is build and pushed above\n",
+    "image_uri = \"{}.dkr.ecr.{}.amazonaws.com/{}:latest\".format(account_id, region_name, repo_name)\n",
+    "base_job_name = 'sd-xl-dreambooth-finetuning-high'\n",
+    "instance_type = 'ml.g5.2xlarge'\n",
+    "inputs = {\n",
+    "    'images': f\"s3://{bucket}/dreambooth-xl/images/\"\n",
+    "}\n",
+    "\n",
+    "estimator = PyTorch(role=role,\n",
+    "                      entry_point='train.sh',\n",
+    "                      source_dir='./sd_xl_dreambooth/',\n",
+    "                      base_job_name=base_job_name,\n",
+    "                      instance_count=1,\n",
+    "                      instance_type=instance_type,\n",
+    "                      image_uri=image_uri,\n",
+    "                      environment=environment,\n",
+    "                      keep_alive_period_in_seconds=3600, #warmpool，为下一次训练保持机器&镜像（滚动续期，最大1hour）；需要开quota。\n",
+    "                      disable_profiler=True,\n",
+    "                      debugger_hook_config=False,\n",
+    "                      max_run=24*60*60*2)\n",
+    "\n",
+    "estimator.fit(inputs)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 57,
+   "id": "bd181b4e-f435-4dca-842a-444d083fdf3c",
+   "metadata": {
+    "pycharm": {
+     "name": "#%%\n"
+    },
+    "tags": []
+   },
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Model artifact saved at:\n",
+      " s3://sagemaker-us-west-2-687912291502/stable-diffusion/dreambooth/\n"
+     ]
+    }
+   ],
+   "source": [
+    "print(\"Model artifact saved at:\\n\", dreambooth_s3uri)"
+   ]
+  }
+ ],
+ "metadata": {
+  "instance_type": "ml.m5.large",
+  "kernelspec": {
+   "display_name": "conda_amazonei_pytorch_latest_p37",
+   "language": "python",
+   "name": "conda_amazonei_pytorch_latest_p37"
+  },
+  "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.7.10"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}