Skip to content

Commit 5e43372

Browse files
committed
upped max_steps in v1-finetune.yaml and fixed TI docs to address CompVis#493
1 parent 7708f4f commit 5e43372

File tree

2 files changed

+18
-8
lines changed

2 files changed

+18
-8
lines changed

configs/stable-diffusion/v1-finetune.yaml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,5 +105,6 @@ lightning:
105105

106106
trainer:
107107
benchmark: True
108-
max_steps: 4000
108+
max_steps: 4000000
109+
# max_steps: 4000
109110

docs/features/TEXTUAL_INVERSION.md

Lines changed: 16 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -18,21 +18,30 @@ To train, prepare a folder that contains images sized at 512x512 and execute the
1818
--init_word 'cat'
1919
```
2020

21-
During the training process, files will be created in /logs/[project][time][project]/
22-
where you can see the process.
21+
During the training process, files will be created in
22+
/logs/[project][time][project]/ where you can see the process.
2323

24-
Conditioning contains the training prompts
25-
inputs, reconstruction the input images for the training epoch samples, samples scaled for a sample of the prompt and one with the init word provided.
24+
Conditioning contains the training prompts inputs, reconstruction the
25+
input images for the training epoch samples, samples scaled for a
26+
sample of the prompt and one with the init word provided.
2627

2728
On a RTX3090, the process for SD will take ~1h @1.6 iterations/sec.
2829

29-
_Note_: According to the associated paper, the optimal number of images is 3-5. Your model may not converge if you use more images than that.
30+
_Note_: According to the associated paper, the optimal number of
31+
images is 3-5. Your model may not converge if you use more images than
32+
that.
3033

31-
Training will run indefinitely, but you may wish to stop it before the heat death of the universe, when you find a low loss epoch or around ~5000 iterations.
34+
Training will run indefinitely, but you may wish to stop it (with
35+
ctrl-c) before the heat death of the universe, when you find a low
36+
loss epoch or around ~5000 iterations. Note that you can set a fixed
37+
limit on the number of training steps by decreasing the "max_steps"
38+
option in configs/stable_diffusion/v1-finetune.yaml (currently set to
39+
4000000)
3240

3341
**Running**
3442

35-
Once the model is trained, specify the trained .pt or .bin file when starting dream using
43+
Once the model is trained, specify the trained .pt or .bin file when
44+
starting dream using
3645

3746
```
3847
(ldm) ~/stable-diffusion$ python3 ./scripts/dream.py --embedding_path /path/to/embedding.pt --full_precision

0 commit comments

Comments
 (0)