28 Commits (5eee2ac39863f9e44591b50d0710dd2615416a13)

Author SHA1 Message Date
Max Audron 5eee2ac398 add data-dir flag and set all user data directories based on it 3 years ago
AUTOMATIC 5228ec8bda remove fairscale requirement, add fake fairscale to make BLIP not complain about it mk2 3 years ago
AUTOMATIC 28189985e6 remove fairscale requirement, add fake fairscale to make BLIP not complain about it 3 years ago
AUTOMATIC 5c1cb9263f fix BLIP failing to import depending on configuration 3 years ago
Vladimir Mandic 04a561c11c
add option to skip interrogate categories 3 years ago
Vladimir Mandic 925dd09c91
improve interrogate 3 years ago
AUTOMATIC 6d805b669e make CLIP interrogator download original text files if the directory does not exist
remove random artist built-in extension (to re-added as a normal extension on demand)
remove artists.csv (but what does it mean????????????????????)
make interrogate buttons show Loading... when you click them
3 years ago
Vladimir Mandic d8d206c168
add state to interrogate 3 years ago
Vladimir Mandic 65be1df7bb
initialize result so not to cause exception on empty results 3 years ago
Yuval Aboulafia 3bf5591efe fix F541 f-string without any placeholders 3 years ago
AUTOMATIC 4b0dc206ed use modelloader for #4956 3 years ago
AUTOMATIC1111 2a649154ec
Merge pull request #4956 from TiagoSantos81/offline_BLIP
[CLIP interrogator] use local file, if available
3 years ago
brkirch 4d5f1691dd Use devices.autocast instead of torch.autocast 3 years ago
Tiago F. Santos a2ae5a6555 [interrogator] mkdir check 3 years ago
Tiago F. Santos 745f1e8f80 [CLIP interrogator] use local file, if available 3 years ago
mawr d587586d3b Added "--clip-models-path" switch to avoid using default "~/.cache/clip" and enable to run under unprivileged user without homedir 3 years ago
Patryk Wychowaniec 7157e5d064 interrogate: Fix CLIP-interrogation on CPU
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).
3 years ago
AUTOMATIC b99d3cf6dd make CLIP interrogate ranks output sane values 3 years ago
DenkingOfficial 58f3ef7733 Fix CLIP Interrogator and disable ranks for it 3 years ago
brkirch fdef8253a4 Add 'interrogate' and 'all' choices to --use-cpu
* Add 'interrogate' and 'all' choices to --use-cpu
* Change type for --use-cpu argument to str.lower, so that choices are case insensitive
3 years ago
Greg Fuller d717eb079c Interrogate: add option to include ranks in output
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 6ed4faac46)
3 years ago
Aidan Holland 432782163a chore: Fix typos 3 years ago
AUTOMATIC c7543d4940 preprocessing for textual inversion added 3 years ago
AUTOMATIC d4dc4c1c63 fix for Error Interrogating when user has garbage in text files #479 3 years ago
AUTOMATIC c50fa7a932 remove mistaken error message 3 years ago
AUTOMATIC 9bb20be090 memory optimization for CLIP interrogator
changed default cfg_scale to a higher value
3 years ago
AUTOMATIC 8fb9c57ed6 add half() supporrt for CLIP interrogation 3 years ago
AUTOMATIC f194457229 CLIP interrogator 3 years ago