Compare commits

...

2 Commits

Author SHA1 Message Date
234ef6e2a1 Update 'ldm/data/personalized.py' 2022-09-29 15:03:02 +02:00
d5d8910541 Update 'dreambooth_runpod_joepenna.ipynb' 2022-09-29 15:02:31 +02:00
2 changed files with 4 additions and 4 deletions

View File

@ -98,7 +98,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"source": "#@markdown Add here the URLs to the images of the concept you are adding\nurls = [\n\"https://prodesk.home.thijn.ovh/gijs/IMG_20220926_121625.png\",\n\"https://prodesk.home.thijn.ovh/gijs/IMG_20220926_121630.png\",\n\"https://prodesk.home.thijn.ovh/gijs/IMG_20220926_121632.png\",\n\"https://prodesk.home.thijn.ovh/gijs/IMG_20220926_121636.png\",\n\"https://prodesk.home.thijn.ovh/gijs/IMG_20220926_121637.png\",\n\"https://prodesk.home.thijn.ovh/gijs/IMG_20220926_121640.png\",\n\"https://prodesk.home.thijn.ovh/gijs/IMG_20220926_121642.png\",\n\"https://prodesk.home.thijn.ovh/gijs/IMG_20220926_121644.png\",\n\"https://prodesk.home.thijn.ovh/gijs/IMG_20220926_121647.png\",\n\"https://prodesk.home.thijn.ovh/gijs/IMG_20220926_121649.png\",\n\"https://prodesk.home.thijn.ovh/gijs/IMG_20220926_121653.png\",\n\"https://prodesk.home.thijn.ovh/gijs/IMG_20220926_121656.png\",\n\"https://prodesk.home.thijn.ovh/gijs/IMG_20220926_121659.png\",\n\"https://prodesk.home.thijn.ovh/gijs/IMG_20220926_121702.png\"\n ## You can add additional images here\n]", "source": "#@markdown Add here the URLs to the images of the concept you are adding\nurls = [\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0133.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0134.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0135.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0136.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0137.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0138.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0139.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0140.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0141.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0142.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0143.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0144.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0145.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0146.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0147.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0148.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0149.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0150.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0151.png\",\n\"https://prodesk.home.thijn.ovh/thijn/IMG_0152.png\"\n ## You can add additional images here\n]",
"metadata": { "metadata": {
"collapsed": false, "collapsed": false,
"jupyter": { "jupyter": {
@ -132,7 +132,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"source": "# START THE TRAINING\nproject_name = \"gijsbert\"\nbatch_size = 1000\nclass_word = \"man\" # << match this word to the class word from regularization images above\nreg_data_root = \"/workspace/Dreambooth-Stable-Diffusion/outputs/txt2img-samples/samples/\" + dataset\n\n!rm -rf training_samples/.ipynb_checkpoints\n!python \"main.py\" \\\n --base configs/stable-diffusion/v1-finetune_unfrozen.yaml \\\n -t \\\n --actual_resume \"model.ckpt\" \\\n --reg_data_root {reg_data_root} \\\n -n {project_name} \\\n --gpus 0, \\\n --data_root \"/workspace/Dreambooth-Stable-Diffusion/training_samples\" \\\n --batch_size {batch_size} \\\n --class_word class_word", "source": "# START THE TRAINING\nproject_name = \"thijngeurts\"\nbatch_size = 1000\nclass_word = \"man\" # << match this word to the class word from regularization images above\nreg_data_root = \"/workspace/Dreambooth-Stable-Diffusion/outputs/txt2img-samples/samples/\" + dataset\n\n!rm -rf training_samples/.ipynb_checkpoints\n!python \"main.py\" \\\n --base configs/stable-diffusion/v1-finetune_unfrozen.yaml \\\n -t \\\n --actual_resume \"model.ckpt\" \\\n --reg_data_root {reg_data_root} \\\n -n {project_name} \\\n --gpus 0, \\\n --data_root \"/workspace/Dreambooth-Stable-Diffusion/training_samples\" \\\n --batch_size {batch_size} \\\n --class_word class_word",
"metadata": { "metadata": {
"id": "6fa5dd66-2ca0-4819-907e-802e25583ae6", "id": "6fa5dd66-2ca0-4819-907e-802e25583ae6",
"tags": [] "tags": []
@ -202,7 +202,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"source": "!echo python scripts/stable_txt2img.py \\\n --ddim_eta 0.0 \\\n --n_samples 1 \\\n --n_iter 4 \\\n --scale 7.0 \\\n --ddim_steps 50 \\\n --ckpt \"/workspace/Dreambooth-Stable-Diffusion/trained_models/\" + {file_name} \\\n --prompt \"gijsbert person as a masterpiece portrait painting by John Singer Sargent in the style of Rembrandt\"", "source": "!echo python scripts/stable_txt2img.py \\\n --ddim_eta 0.0 \\\n --n_samples 1 \\\n --n_iter 4 \\\n --scale 7.0 \\\n --ddim_steps 50 \\\n --ckpt \"/workspace/Dreambooth-Stable-Diffusion/trained_models/\" + {file_name} \\\n --prompt \"thijngeurts person as a masterpiece portrait painting by John Singer Sargent in the style of Rembrandt\"",
"metadata": { "metadata": {
"trusted": true "trusted": true
}, },

View File

@ -8,7 +8,7 @@ from torchvision import transforms
import random import random
training_templates_smallest = [ training_templates_smallest = [
'gijsbert {}', 'thijngeurts {}',
] ]
reg_templates_smallest = [ reg_templates_smallest = [