Compare commits
2 Commits
Author | SHA1 | Date | |
---|---|---|---|
234ef6e2a1 | |||
d5d8910541 |
@ -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
|
||||||
},
|
},
|
||||||
|
@ -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 = [
|
||||||
|
Loading…
Reference in New Issue
Block a user