Commit Graph

137 Commits

Author SHA1 Message Date
xra
fef632e0e1 tokenization logging (take 2)
This adds an option -t argument that will print out color-coded tokenization, SD has a maximum of 77 tokens, it silently discards tokens over the limit if your prompt is too long.
By using -t you can see how your prompt is being tokenized which helps prompt crafting.
2022-08-29 12:28:49 +09:00
05061a70b3 report errors on non-cuda systems rather than failing silently 2022-08-28 23:13:23 -04:00
617a029ae7 pass outdir from txt2img() and img2img() to prompt2img() correctly 2022-08-28 23:12:49 -04:00
8a517fdc17 Fix sampler changer not working. 2022-08-29 11:26:19 +12:00
b85773f332 resolved conflicts and write properly-formatted prompt string (with sampler & upscaling) into image file 2022-08-28 19:01:45 -04:00
8bbe7936bd close Issue #165 2022-08-28 18:21:20 -04:00
7c485a1a4a adjusted -U upscaling argument so that it defaults to upscaling strength 0.75 if the second argument is not given 2022-08-28 17:26:39 -04:00
6e49c070bb Optimize and Improve GFPGAN and Real-ESRGAN Pipeline 2022-08-29 08:14:29 +12:00
a7ac93a899 Merge pull request #110 from sajattack/half-precision-embeddings
Support full-precision embeddings in half precision inference mode
2022-08-28 15:36:26 -04:00
3b2569ebdd Merge branch 'yunsaki-main' into main 2022-08-28 14:20:48 -04:00
8b9a520c5c adjusted handling of from_file 2022-08-28 14:20:34 -04:00
ba03289c14 print current and max VRAM usage stats after each round of generation 2022-08-28 13:05:01 -04:00
d1551b1bd4 Enable users to set sampler using prompts 2022-08-29 04:27:54 +12:00
800132970e Merge pull request #105 from shusso/select-device
Move torch.device selection to it's own function
2022-08-26 12:23:21 -04:00
555f13e469 Merge branch 'main' into half-precision-embeddings 2022-08-26 08:33:46 -07:00
9b5101cd8d support full-precision embeddings in half precision mode 2022-08-26 08:30:58 -07:00
5129f256a3 simplet2i: changed image file handling to work as stated in the [docs](https://pillow.readthedocs.io/en/stable/reference/open_files.html) 2022-08-26 14:13:16 +02:00
ed72ff3268 Move torch.device selection to it's own function 2022-08-26 14:43:18 +03:00
f3fad22cb6 Fix 2022-08-26 05:27:34 -04:00
7bf0bc5208 fix comment 2022-08-26 04:08:18 -04:00
4e5aa7e714 fix comment 2022-08-26 04:07:01 -04:00
46a223f229 Double check for null and 0, and add a comment to indicate intent 2022-08-26 04:05:09 -04:00
4f02b72c9c prettified all the code using "blue" at the urging of @tildebyte 2022-08-26 03:15:42 -04:00
407d70a987 Fix backwards logic 2022-08-26 00:49:12 -04:00
f1ffb5b51b Fix blend if the target image has been upscaled 2022-08-26 00:45:19 -04:00
4f1664ec4f remove params 2022-08-26 00:41:41 -04:00
fcdd95b652 Refactor so that behavior is consolidated at top level 2022-08-26 00:39:57 -04:00
470a62dbbe Merge branch 'main' of https://github.com/BaristaLabs/stable-diffusion-dream into add-gfpgan-option 2022-08-26 00:26:03 -04:00
cb86b9ae6e Remove the redundancy, better logging 2022-08-25 23:48:35 -04:00
dbb9132f4d Merge branch 'main' of https://github.com/BaristaLabs/stable-diffusion-dream into add-gfpgan-option 2022-08-25 23:19:17 -04:00
5711b6d611 Add optional GFPGAN support 2022-08-25 22:57:30 -04:00
f1bed52530 moved dream utilities into their own subfolder 2022-08-25 22:49:15 -04:00
c38b6964b4 improved inline error messages slightly 2022-08-25 22:19:12 -04:00
797de3257c fix batch_size 2022-08-25 17:28:52 -07:00
31b22e057d switch to generators 2022-08-25 17:06:06 -07:00
078859207d factor out loop 2022-08-25 16:51:39 -07:00
a10baf5808 factor out exception handler 2022-08-25 15:13:07 -07:00
19fa222810 refactoring complete; please test carefully! 2022-08-25 17:30:08 -04:00
b3e3b0e861 feature complete; looks like ready for merge 2022-08-25 17:26:48 -04:00
650ae3eb13 Merge pull request #89 from BlueAmulet/remove-accelerate
Remove accelerate library
2022-08-25 13:48:48 -04:00
39b55ae016 Remove accelerate library
This library is not required to use k-diffusion
Make k-diffusion wrapper closer to the other samplers
2022-08-25 11:04:57 -06:00
e82c5eba18 PR revision: replace cuda call with dynamic type 2022-08-25 12:18:35 -04:00
49247b4aa4 fix performance regression; closes issue #42 2022-08-25 09:41:12 -04:00
eb58276a2c fix VRAM utilization 2022-08-25 08:34:51 -04:00
0b4459b707 mostly back to full functionality; just missing grid generation code 2022-08-25 00:42:37 -04:00
b978536385 code is reorganized and mostly functional. Grid needs to be brought back online, as well as naming of img2img variants (currently the variants get written but not logged) 2022-08-24 19:47:59 -04:00
9133087850 first draft at big refactoring; will be broken 2022-08-24 17:52:34 -04:00
1eec6b776b tweaked documentation and comments slightly 2022-08-24 15:25:52 -04:00
feea5fb063 Merge branch 'main' of https://github.com/BaristaLabs/stable-diffusion-dream into add-simple-variant-mechanism 2022-08-24 13:16:15 -04:00
b5cdbd3b0b Fixes issue with cuda/current mismatch 2022-08-24 13:14:08 -04:00