-
Notifications
You must be signed in to change notification settings - Fork 0
/
run.sh
49 lines (43 loc) · 2.24 KB
/
run.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
## Change Jack to Simon using mixed precision training and prior preservation
# python illustrator.py \
# --orig_name Jack \
# --custom_name Simon \
# --custom_img_dir "sample_images/simon_512" \
# --prompts_path output/jack_and_the_beanstalk/story.txt \
# --config_path "config/text-encoder-mixed-precision-prior-preserve.yml" \
# --device cuda:0 \
## change jack to simon but with generic orig_name, specific custom_name, and generic character description
python illustrator.py \
--orig_object "boy" \
--orig_name "Jack" \
--custom_name Simonstasta \
--custom_img_dir "sample_images/simon_512" \
--prompts_path output/jack_and_the_beanstalk/story.txt \
--config_path "config/text-encoder-mixed-precision-prior-preserve.yml" \
--device cuda:0 \
--skip_training
## load a previously trained dreambooth model
# python illustrator.py \
# --orig_name Jack \
# --prompts_path output/jack_and_the_beanstalk/story.txt \
# --config_path "config/dreambooth-sd1-5.yml" \
# --device cuda:0 \
# --model_id "runs/dreambooth-model"
## mixed precision inference using model without mixed precision: works very well without error
# python illustrator.py \
# --orig_name Jack \
# --prompts_path output/jack_and_the_beanstalk/story.txt \
# --prefix "mdjrny-v4 kids illustration showcasing the story of 'Jack and the Beanstalk'" \
# --suffix "drawn by Rebecca Sugar, bright engaging children's illustration, digital painting, big eyes, beautiful shading, beautiful colors, amazon kdp, happy, interesting, 2D" \
# --config_path "config/text-encoder-mixed-precision.yml" \
# --device cuda:0 \
# --model_id "runs/dreambooth-model"
# load mixed precision trained model
# python illustrator.py \
# --orig_name Jack \
# --prompts_path output/jack_and_the_beanstalk/story.txt \
# --prefix "mdjrny-v4 kids illustration showcasing the story of 'Jack and the Beanstalk'" \
# --suffix "drawn by Rebecca Sugar, bright engaging children's illustration, digital painting, big eyes, beautiful shading, beautiful colors, amazon kdp, happy, interesting, 2D" \
# --config_path "config/text-encoder-mixed-precision.yml" \
# --device cuda:0 \
# --model_id "runs/text-encoder-mixed-precision"