Commit Graph

8 Commits

Author SHA1 Message Date
Lizardy
62790406bf
Fix --config args
* Fixed config args being double-loaded and not replacing the defaults.
2022-09-26 23:58:41 -04:00
hlky
4a15702fe9
License header 2022-09-26 14:02:48 +01:00
Chris Heald
c7e126f6f8
Updates for the v2 trainer
* Restore --config. This will be useful when you have an init config
  that you don't want overwritten.
* Cache the individual transformed images in TextualInversionDataset.
  This gains speed by avoiding reading and reprocessing the image each
  time it's used for training.
* Turn on no_grad for inference and clean up tensors during
  checkpointing. This reduces memory usage slightly.
* Set the sample output size to 384x384. We just need them large enough
  for manual evaluation, and this gains us a decent chunk of speed.
* (breaking change) Custom templates are now semicolon-delineated.
  Additionally, custom templates are properly passed through to
  TextualInversionDataset to generate input_ids for your images. Using
  custom templates which accurately describe your input images seems to
  improve training fidelity.
* Cache autoencoding of image pixel data. This substantially increases
  the speed of training, upwards of 40% for me.
* Clean up a little bit of cruft.
2022-09-25 13:21:24 +01:00
hlky
0d79243e45
fix diffusers_textual_inversion_2.py conflicts step 1 2022-09-25 13:20:17 +01:00
ZeroCool940711
7084e94bdf Moved the model to server_state instead of session_state so it can be shared between multiple sessions, tabs and users as long as the streamlit server is running. 2022-09-25 00:03:05 -07:00
hlky
8b4464a706
Add periodic checkpointing
Update diffusers_textual_inversion_2.py
2022-09-23 15:29:08 +01:00
hlky
c6bf0d2ecb
Update diffusers_textual_inversion_2.py 2022-09-19 14:49:00 +01:00
hlky
91c0d0f20b
Create diffusers_textual_inversion_2.py 2022-09-19 13:09:45 +01:00