aboutsummaryrefslogtreecommitdiff
path: root/modules/interrogate.py
AgeCommit message (Collapse)Author
2022-12-03use modelloader for #4956AUTOMATIC
2022-12-03Merge pull request #4956 from TiagoSantos81/offline_BLIPAUTOMATIC1111
[CLIP interrogator] use local file, if available
2022-11-30Use devices.autocast instead of torch.autocastbrkirch
2022-11-24[interrogator] mkdir checkTiago F. Santos
2022-11-22[CLIP interrogator] use local file, if availableTiago F. Santos
2022-10-31Added "--clip-models-path" switch to avoid using default "~/.cache/clip" and ↵mawr
enable to run under unprivileged user without homedir
2022-10-21interrogate: Fix CLIP-interrogation on CPUPatryk Wychowaniec
Currently, trying to perform CLIP interrogation on a CPU fails, saying: ``` RuntimeError: "slow_conv2d_cpu" not implemented for 'Half' ``` This merge request fixes this issue by detecting whether the target device is CPU and, if so, force-enabling `--no-half` and passing `device="cpu"` to `clip.load()` (which then does some extra tricks to ensure it works correctly on CPU).
2022-10-17make CLIP interrogate ranks output sane valuesAUTOMATIC
2022-10-17Fix CLIP Interrogator and disable ranks for itDenkingOfficial
2022-10-14Add 'interrogate' and 'all' choices to --use-cpubrkirch
* Add 'interrogate' and 'all' choices to --use-cpu * Change type for --use-cpu argument to str.lower, so that choices are case insensitive
2022-10-11Interrogate: add option to include ranks in outputGreg Fuller
Since the UI also allows users to specify ranks, it can be useful to show people what ranks are being returned by interrogate This can also give much better results when feeding the interrogate results back into either img2img or txt2img, especially when trying to generate a specific character or scene for which you have a similar concept image Testing Steps: Launch Webui with command line arg: --deepdanbooru Navigate to img2img tab, use interrogate DeepBooru, verify tags appears as before. Use "Interrogate CLIP", verify prompt appears as before Navigate to Settings tab, enable new option, click "apply settings" Navigate to img2img, Interrogate DeepBooru again, verify that weights appear and are properly formatted. Note that "Interrogate CLIP" prompt is still unchanged In my testing, this change has no effect to "Interrogate CLIP", as it seems to generate a sentence-structured caption, and not a set of tags. (reproduce changes from https://github.com/AUTOMATIC1111/stable-diffusion-webui/pull/2149/commits/6ed4faac46c45ca7353f228aca9b436bbaba7bc7)
2022-10-08chore: Fix typosAidan Holland
2022-10-02preprocessing for textual inversion addedAUTOMATIC
2022-09-15fix for Error Interrogating when user has garbage in text files #479AUTOMATIC
2022-09-12remove mistaken error messageAUTOMATIC
2022-09-12memory optimization for CLIP interrogatorAUTOMATIC
changed default cfg_scale to a higher value
2022-09-11add half() supporrt for CLIP interrogationAUTOMATIC
2022-09-11CLIP interrogatorAUTOMATIC