{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "32291dd5-3b3a-4b19-ab79-508859e43cb0",
   "metadata": {
    "collapsed": true,
    "jupyter": {
     "outputs_hidden": true
    },
    "tags": []
   },
   "outputs": [],
   "source": [
    "!git clone https://github.com/XavierXiao/Dreambooth-Stable-Diffusion.git"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "4eb45ec3-f5c6-424f-9a15-ae0569bbedc3",
   "metadata": {
    "collapsed": true,
    "jupyter": {
     "outputs_hidden": true
    },
    "tags": []
   },
   "outputs": [],
   "source": [
    "cd Dreambooth-Stable-Diffusion"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "cde27f84-66e5-4da0-8f01-b99a70f95145",
   "metadata": {
    "collapsed": true,
    "jupyter": {
     "outputs_hidden": true
    },
    "tags": []
   },
   "outputs": [],
   "source": [
    "pip install -e ."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "19512b6c-c021-4a46-bee2-8d9af0a1f4a0",
   "metadata": {},
   "outputs": [],
   "source": [
    "mkdir inputs reg_inputs outputs checkpoint"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "b7806869-da36-400e-942c-c8a83de74126",
   "metadata": {
    "collapsed": true,
    "jupyter": {
     "outputs_hidden": true
    },
    "tags": []
   },
   "outputs": [],
   "source": [
    "pip install gdown"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "0b3137e6-6d81-4042-bd7c-fc4df6fa4644",
   "metadata": {
    "collapsed": true,
    "jupyter": {
     "outputs_hidden": true
    },
    "tags": []
   },
   "outputs": [],
   "source": [
    "!gdown https://drive.google.com/uc?id=1bo5mB_gw8pO-R6Wx1M6EL5ap0llHiYmA"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "dc465bf7-669a-4ab9-97aa-a99b98d355c0",
   "metadata": {
    "collapsed": true,
    "jupyter": {
     "outputs_hidden": true
    },
    "tags": []
   },
   "outputs": [],
   "source": [
    "!python /workspace/Dreambooth-Stable-Diffusion/setup.py install"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "0623663f-abef-4038-bc19-0b3fbd3103fc",
   "metadata": {
    "collapsed": true,
    "jupyter": {
     "outputs_hidden": true
    },
    "tags": []
   },
   "outputs": [],
   "source": [
    "pip install pytorch-lightning==1.6.5"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "2a91ee93-a521-4e5b-9761-4905d9b30d06",
   "metadata": {
    "collapsed": true,
    "jupyter": {
     "outputs_hidden": true
    },
    "tags": []
   },
   "outputs": [],
   "source": [
    "pip install omegaconf"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3b3be41b-019c-4820-bbdd-c89bca727365",
   "metadata": {
    "collapsed": true,
    "jupyter": {
     "outputs_hidden": true
    },
    "tags": []
   },
   "outputs": [],
   "source": [
    "pip install einops"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "fac502b3-5dfa-4eca-a6d8-adc90bb81553",
   "metadata": {
    "collapsed": true,
    "jupyter": {
     "outputs_hidden": true
    },
    "tags": []
   },
   "outputs": [],
   "source": [
    "pip install taming-transformers"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "c0cc2ebc-9248-48b2-b271-52cc8da48723",
   "metadata": {
    "collapsed": true,
    "jupyter": {
     "outputs_hidden": true
    },
    "tags": []
   },
   "outputs": [],
   "source": [
    "pip install clip"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "64b78c02-a7b5-42e3-8216-55e017295984",
   "metadata": {
    "collapsed": true,
    "jupyter": {
     "outputs_hidden": true
    },
    "tags": []
   },
   "outputs": [],
   "source": [
    "pip install transformers"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "5a0bfcc5-0531-4ada-8bf1-1e9b36e8bdc4",
   "metadata": {
    "collapsed": true,
    "jupyter": {
     "outputs_hidden": true
    },
    "tags": []
   },
   "outputs": [],
   "source": [
    "pip install kornia"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "ba5765cb-1a6b-4b0a-87bd-fbb81693ba2b",
   "metadata": {
    "collapsed": true,
    "jupyter": {
     "outputs_hidden": true
    },
    "tags": []
   },
   "outputs": [],
   "source": [
    "pip install test-tube"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3f2a0105-fcba-42f1-a0c6-da1860fa3f7d",
   "metadata": {},
   "outputs": [],
   "source": [
    "test"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "211c7a87-a5a6-4ca6-9f35-e591e8db1261",
   "metadata": {},
   "outputs": [],
   "source": [
    "!python /workspace/Dreambooth-Stable-Diffusion/main.py --no-test --base configs/stable-diffusion/v1-finetune_unfrozen.yaml -t --actual_resume ./checkpoint/NovelAI.ckpt -n 1girl --gpus 0, --data_root ./inputs --reg_data_root ./reg_inputs --class_word 1girl"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "9c1ec741-373c-4e3a-84a1-175f7a868fc2",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.13"
  },
  "toc-autonumbering": true
 },
 "nbformat": 4,
 "nbformat_minor": 5
}