Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

changed regularization image mv command to jpg #147

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

changed regularization image mv command to jpg #147

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
72 changes: 36 additions & 36 deletions dreambooth_runpod_joepenna.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -48,6 +48,9 @@
{
"cell_type": "code",
"execution_count": null,
"metadata": {
"collapsed": false
},
"outputs": [],
"source": [
"# BUILD ENV\n",
Expand All @@ -68,20 +71,17 @@
"!pip install -qq diffusers[\"training\"]==0.3.0 transformers ftfy\n",
"!pip install huggingface_hub\n",
"!pip install captionizer==1.0.1\n"
],
"metadata": {
"collapsed": false
}
]
},
{
"cell_type": "markdown",
"metadata": {
"collapsed": false
},
"source": [
"# Hugging Face Token\n",
"Grab your hugging face token from here https://huggingface.co/settings/tokens and paste into the next cell (HUGGING_FACE_API_TOKEN_HERE)"
],
"metadata": {
"collapsed": false
}
]
},
{
"cell_type": "code",
Expand All @@ -98,17 +98,20 @@
},
{
"cell_type": "markdown",
"metadata": {
"collapsed": false
},
"source": [
"# Download the 1.5 model from hugging face\n",
"You can also provide your own v1.* model for training by uploading it and renaming it to \"model.ckpt\". It should be in the same directory as dreambooth_runpod_joepenna.ipynb"
],
"metadata": {
"collapsed": false
}
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {
"collapsed": false
},
"outputs": [],
"source": [
"# Download the 1.5 sd model\n",
Expand All @@ -126,10 +129,7 @@
"!mv {actual_locations_of_model_blob[-1]} model.ckpt\n",
"clear_output()\n",
"print(\"✅ model.ckpt successfully downloaded\")"
],
"metadata": {
"collapsed": false
}
]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -181,7 +181,7 @@
"dataset=self_generated_files_prompt\n",
"\n",
"!mkdir -p regularization_images/{dataset}\n",
"!mv outputs/txt2img-samples/*.png regularization_images/{dataset}"
"!mv outputs/txt2img-samples/*.jpg regularization_images/{dataset}"
]
},
{
Expand All @@ -201,6 +201,9 @@
},
{
"cell_type": "markdown",
"metadata": {
"collapsed": false
},
"source": [
"# Download pre-generated regularization images\n",
"We've created the following image sets\n",
Expand All @@ -211,10 +214,7 @@
"`person_ddim`\n",
"`woman_ddim` - provided by David Bielejeski - ddim @ 50 steps, CFG 10.0\n",
"`person_ddim` is recommended"
],
"metadata": {
"collapsed": false
}
]
},
{
"cell_type": "code",
Expand All @@ -236,6 +236,9 @@
},
{
"cell_type": "markdown",
"metadata": {
"collapsed": false
},
"source": [
"# Upload your training images\n",
"Upload 10-20 images of someone to\n",
Expand All @@ -253,14 +256,14 @@
"The images should be:\n",
"\n",
"- as close as possible to the kind of images you're trying to make"
],
"metadata": {
"collapsed": false
}
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {
"collapsed": false
},
"outputs": [],
"source": [
"#@markdown Add here the URLs to the images of the subject you are adding\n",
Expand All @@ -272,10 +275,7 @@
" \"https://i.imgur.com/test5.png\",\n",
" # You can add additional images here -- about 20-30 images in different\n",
"]"
],
"metadata": {
"collapsed": false
}
]
},
{
"cell_type": "code",
Expand Down Expand Up @@ -452,17 +452,17 @@
},
{
"cell_type": "markdown",
"metadata": {
"collapsed": false
},
"source": [
"# Optional - Upload to google drive\n",
"* run the following commands in a new `terminal` in the `Dreambooth-Stable-Diffusion` directory\n",
"* `chmod +x ./gdrive`\n",
"* `./gdrive about`\n",
"* `paste your token here after navigating to the link`\n",
"* `./gdrive upload trained_models/{file_name.ckpt}`"
],
"metadata": {
"collapsed": false
}
]
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -506,7 +506,7 @@
"provenance": []
},
"kernelspec": {
"display_name": "Python 3 (ipykernel)",
"display_name": "Python 3.8.3 ('base')",
"language": "python",
"name": "python3"
},
Expand All @@ -520,11 +520,11 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.10.6"
"version": "3.8.3"
},
"vscode": {
"interpreter": {
"hash": "b0fa6594d8f4cbf19f97940f81e996739fb7646882a419484c72d19e05852a7e"
"hash": "99cb154694ef233467f574b5f2483eef2931fc8692c01c95dc679d143515eea0"
}
}
},
Expand Down